From 6374657f2aff683d922974a4b801df0bfcb28688 Mon Sep 17 00:00:00 2001 From: gaotingquan Date: Fri, 7 Jul 2023 07:45:52 +0000 Subject: [PATCH] support to specify save dir --- deploy/python/predict_system.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/deploy/python/predict_system.py b/deploy/python/predict_system.py index 19bf5c325..cb6cd4d79 100644 --- a/deploy/python/predict_system.py +++ b/deploy/python/predict_system.py @@ -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