mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

support to specify save dir

Browse Source
This commit is contained in:
gaotingquan 2023-07-07 07:45:52 +00:00 committed by Tingquan Gao
parent bf865cb0d8
commit 6374657f2a
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
deploy/python/predict_system.py
View File

@ -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