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

support to specify save dir

上级 48fc521e
# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
......@@ -144,7 +144,8 @@ def main(config):
for idx, image_file in enumerate(image_list):
img = cv2.imread(image_file)[:, :, ::-1]
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)
return
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册