diff --git a/deploy/TensorRT/README.md b/deploy/TensorRT/README.md index 960f7d6d..4057acc4 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.40 --iou-thres 0.45 +```