Linaom1214/TensorRT-For-YOLO-Series

yolov7，official，int8，onnx-> trt报错

simplew2011 opened this issue · 3 comments

simplew2011 commented 
wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt
python export.py --weights ./yolov7.pt --grid --end2end --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640
git clone https://github.com/Linaom1214/tensorrt-python.git
python ./tensorrt-python/export.py -o yolov7.onnx -e yolov7-nms.trt -p fp16  ---> ok
python ./tensorrt-python/export.py -o yolov7.onnx -e yolov7-nms-int8.trt -p int8 --calib_input /home/vastai/nvbenchmark/data/detection/det_coco_calib --end2end --conf_thres 0.25 --iou_thres 0.45  ---> error
simplew2011 commented

Affected by --end2end ?

Linaom1214 commented

请参考本项目最新的文档