insight-platform / Savant

Python Computer Vision & Video Analytics Framework With Batteries Included
https://savant-ai.io
Apache License 2.0
510 stars 41 forks source link

Trying to export my custom yolov7 model from .pt to .onnx #796

Closed chanwitkepha closed 1 week ago

chanwitkepha commented 1 week ago

Dear all, I'm trying to export my custom yolov7 model from .pt to .onnx to be used in Savant. Should I export with nms included or not? It seems like the yolo converter in Savant doesn’t handle the case with nms included.

Thank you in advance for your help.

bwsw commented 1 week ago

Please ask use case questions on Discord. Here we mostly handle bugs, complex cases and features.