提交 8ed20600 编写于 作者: G gaotingquan 提交者: Tingquan Gao

support to specify save dir

上级 48fc521e
...@@ -144,7 +144,8 @@ def main(config): ...@@ -144,7 +144,8 @@ def main(config):
for idx, image_file in enumerate(image_list): for idx, image_file in enumerate(image_list):
img = cv2.imread(image_file)[:, :, ::-1] img = cv2.imread(image_file)[:, :, ::-1]
output = system_predictor.predict(img) output = system_predictor.predict(img)
draw_bbox_results(img, output, image_file) save_dir = config["Global"]["output_dir"]
draw_bbox_results(img, output, image_file, save_dir=save_dir)
print(output) print(output)
return return
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册