Skip to content

M
models

PaddlePaddle / models
接近 2 年 前同步成功

通知 230
Star 6828
Fork 2962
M
models
提交 ffc79784 编写于 作者: S soldier828
浏览文件
操作

fix parameter snapshot_iter

上级 616a8ae7
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
fluid/PaddleCV/rcnn/utility.py
浏览文件 @ ffc79784
...@@ -157,7 +157,7 @@ def parse_args(): ...@@ -157,7 +157,7 @@ def parse_args():
add_arg('pixel_means', float, [102.9801, 115.9465, 122.7717], "pixel mean") add_arg('pixel_means', float, [102.9801, 115.9465, 122.7717], "pixel mean")
add_arg('nms_thresh', float, 0.5, "NMS threshold.") add_arg('nms_thresh', float, 0.5, "NMS threshold.")
add_arg('score_thresh', float, 0.05, "score threshold for NMS.") add_arg('score_thresh', float, 0.05, "score threshold for NMS.")
add_arg('snapshot_stride', int, 10000, "save model every snapshot stride.") add_arg('snapshot_iter', int, 10000, "save model every snapshot iteration.")
# SINGLE EVAL AND DRAW # SINGLE EVAL AND DRAW
add_arg('draw_threshold', float, 0.8, "Confidence threshold to draw bbox.") add_arg('draw_threshold', float, 0.8, "Confidence threshold to draw bbox.")
add_arg('image_path', str, 'dataset/coco/val2017', "The image path used to inference and visualize.") add_arg('image_path', str, 'dataset/coco/val2017', "The image path used to inference and visualize.")
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或