Can this QAT model fit the original YOLOv7's detect.py/test.py ?

NVIDIA-AI-IOT / yolo_deepstream

yolo model qat and deploy with deepstream&tensorrt

Apache License 2.0

549 stars 139 forks source link

Can this QAT model fit the original YOLOv7's detect.py/test.py ? #58

Open RonaldYuren opened 6 months ago

RonaldYuren commented 6 months ago

Just like the title, I want to do QAT the use the original YOLOv7's detect.py/test.py. How to I save the model and then load it by detect.py/test.py ? Would it work if I do not save the model in onnx format ?