From 0a26a94b553a271fa763f9b137fa8364468b61f7 Mon Sep 17 00:00:00 2001 From: Alessio Placitelli Date: Fri, 10 Jan 2025 16:00:20 +0100 Subject: [PATCH] Rename arguments to match code usage `COCO_dir` is not used (`infer_dir` is instead referenced in the code), and `engine_dir` is used in place of `models_dir`. --- tools/benchmark/trt_benchmark.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/benchmark/trt_benchmark.py b/tools/benchmark/trt_benchmark.py index a650ac0..328d283 100644 --- a/tools/benchmark/trt_benchmark.py +++ b/tools/benchmark/trt_benchmark.py @@ -19,7 +19,7 @@ def parse_args(): parser = argparse.ArgumentParser(description='Argument Parser Example') - parser.add_argument('--COCO_dir', + parser.add_argument('--infer_dir', type=str, default='/data/COCO2017/val2017', help="Directory for images to perform inference on.") @@ -180,7 +180,7 @@ def main(): 'orig_target_sizes': torch.tensor([640, 640]).to(im.device), } - engine_files = glob.glob(os.path.join(FLAGS.models_dir, "*.engine")) + engine_files = glob.glob(os.path.join(FLAGS.engine_dir, "*.engine")) results = [] for engine_file in engine_files: