From 2b65cbae7eca55a574d3ed57d7bd15de3f95666a Mon Sep 17 00:00:00 2001 From: Akash A Desai <62583018+akashAD98@users.noreply.github.com> Date: Thu, 15 Sep 2022 13:18:26 +0530 Subject: [PATCH 1/2] updated redme for doing images testing using .trt weight we can do testing --- deploy/TensorRT/README.md | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/deploy/TensorRT/README.md b/deploy/TensorRT/README.md index 960f7d6d..c4829afe 100644 --- a/deploy/TensorRT/README.md +++ b/deploy/TensorRT/README.md @@ -69,4 +69,13 @@ Then run the demo: ```shell ./yolov6 ../you.engine -i image_path -``` \ No newline at end of file +``` +# Testing on image +You can do testing on images using .trt weights, just give path of image directory & its annotation path + +``` +python3 deploy/TensorRT/eval_yolo_trt.py -v -m model.trt \ +--imgs-dir /workdir/datasets/coco/images/val2017 \ +--annotations /workdir/datasets/coco/annotations/instances_val2017.json \ +--conf-thres 0.25 --iou-thres 0.45 +``` From 467c8f6a9f40938bf830568ff7acbfdaa9240379 Mon Sep 17 00:00:00 2001 From: Akash A Desai <62583018+akashAD98@users.noreply.github.com> Date: Thu, 15 Sep 2022 19:31:00 +0530 Subject: [PATCH 2/2] Update README.md --- deploy/TensorRT/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deploy/TensorRT/README.md b/deploy/TensorRT/README.md index c4829afe..4057acc4 100644 --- a/deploy/TensorRT/README.md +++ b/deploy/TensorRT/README.md @@ -77,5 +77,5 @@ You can do testing on images using .trt weights, just give path of image directo python3 deploy/TensorRT/eval_yolo_trt.py -v -m model.trt \ --imgs-dir /workdir/datasets/coco/images/val2017 \ --annotations /workdir/datasets/coco/annotations/instances_val2017.json \ ---conf-thres 0.25 --iou-thres 0.45 +--conf-thres 0.40 --iou-thres 0.45 ```