mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 14:31:29 +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

5
deploy/python/predict_system.py
View File

@ -1,6 +1,6 @@
# 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