NVIDIA-AI-IOT / yolo_deepstream

yolo model qat and deploy with deepstream&tensorrt
Apache License 2.0
533 stars 135 forks source link

yolov7_qat #41

Open junghyun-avikus opened 1 year ago

junghyun-avikus commented 1 year ago

Will yolov7_qat support custom data training instead of COCO in the near future?

wanghr323 commented 1 year ago

Just change the dataset to your own dataset, I think there is no difference between the offical yolov7 with yolov7 qat except adding some QDQ nodes into the model