提交 eb1fbf5c 编写于 作者: L LielinJiang 提交者: qingqing01

Fix deeplabv3p eval bug on windows platform (#3235)

上级 21e406db
......@@ -65,7 +65,7 @@ python ./train.py \
--save_weights_path=$SAVE_WEIGHTS_PATH \
--dataset_path=$DATASET_PATH
```
注意:如果在windows系统上进行训练请不要将use_multiprocessing设置为True
注意:如果在windows系统上进行训练,预测请不要将use_multiprocessing设置为True
使用以下命令获得更多使用说明:
```
......
......@@ -39,6 +39,7 @@ add_arg('dataset_path', str, None, "Cityscape dataset path.")
add_arg('use_gpu', bool, True, "Whether use GPU or CPU.")
add_arg('num_classes', int, 19, "Number of classes.")
add_arg('use_py_reader', bool, True, "Use py_reader.")
add_arg('use_multiprocessing', bool, False, "Use multiprocessing.")
add_arg('norm_type', str, 'bn', "Normalization type, should be 'bn' or 'gn'.")
#yapf: enable
......@@ -124,7 +125,7 @@ if args.total_step == -1:
else:
total_step = args.total_step
batches = dataset.get_batch_generator(batch_size, total_step)
batches = dataset.get_batch_generator(batch_size, total_step, use_multiprocessing=args.use_multiprocessing)
if args.use_py_reader:
py_reader.decorate_tensor_provider(lambda :[ (yield b[0],b[1]) for b in batches])
py_reader.start()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册