yolov7 icon indicating copy to clipboard operation
yolov7 copied to clipboard

shape inference of TRT::EfficientNMS_TRT type is missing

Open Kuchchi12 opened this issue 1 year ago • 1 comments

I trained yolov7 tiny for 4 classes for recognition of face landmarks face, eye, nose and mouth further i am trying to convert .pt --> .onnx and facing this warning Screenshot 2024-04-04 160704 which ig is causing problem in .onnx -->.trt conversion because of not getting desired shape Screenshot 2024-04-04 160825 also it would be great help if anyone can teach me how to decrease model size while conversion

python : Python 3.8.10 torch : 2.2.0+cu118

Kuchchi12 avatar Apr 04 '24 10:04 Kuchchi12

Try this notebook for the conversion- https://colab.research.google.com/gist/AlexeyAB/fcb47ae544cf284eb24d8ad8e880d45c/yolov7trtlinaom.ipynb

dsbyprateekg avatar Apr 05 '24 09:04 dsbyprateekg