Merge pull request meituan#466 from akashAD98/patch-1

updated redme for doing images testing
wobuhuicode · Sep 16, 2022 · 3b87280 · 3b87280
2 parents 124a67c + 467c8f6
commit 3b87280
Showing 1 changed file with 10 additions and 1 deletion.
diff --git 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
-```
+```
+# 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.40 --iou-thres 0.45
+```