提交 84e4cbba 编写于 作者: B Bai Yifan 提交者: qingqing01

update file path (#1137)

上级 7ff5fb00
...@@ -24,7 +24,7 @@ add_arg('skip_reader', bool, False, "Whether to skip data reade ...@@ -24,7 +24,7 @@ add_arg('skip_reader', bool, False, "Whether to skip data reade
add_arg('use_gpu', bool, True, "Whether use GPU.") add_arg('use_gpu', bool, True, "Whether use GPU.")
add_arg('use_pyramidbox', bool, True, "Whether use PyramidBox model.") add_arg('use_pyramidbox', bool, True, "Whether use PyramidBox model.")
add_arg('model_save_dir', str, 'output', "The path to save model.") add_arg('model_save_dir', str, 'output', "The path to save model.")
add_arg('pretrained_model', str, './pretrained/', "The init model path.") add_arg('pretrained_model', str, './vgg_ilsvrc_16_fc_reduced', "The init model path.")
add_arg('resize_h', int, 640, "The resized image height.") add_arg('resize_h', int, 640, "The resized image height.")
add_arg('resize_w', int, 640, "The resized image height.") add_arg('resize_w', int, 640, "The resized image height.")
#yapf: enable #yapf: enable
...@@ -177,8 +177,8 @@ if __name__ == '__main__': ...@@ -177,8 +177,8 @@ if __name__ == '__main__':
args = parser.parse_args() args = parser.parse_args()
print_arguments(args) print_arguments(args)
data_dir = 'data/WIDERFACE/WIDER_train/images/' data_dir = 'data/WIDER_train/images/'
train_file_list = 'label/train_gt_widerface.res' train_file_list = 'data/wider_face_split/wider_face_train_bbx_gt.txt'
config = reader.Settings( config = reader.Settings(
data_dir=data_dir, data_dir=data_dir,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册