search

NVIDIA-AI-IOT / cuDLA-samples

YOLOv5 on Orin DLA
Other
167 stars 17 forks source link

Quantization of yolov5 #11

Open zifeiyu-tan opened 6 months ago

zifeiyu-tan commented 6 months ago

If i only want to quantize with int8 in yolov5, how to change this code。

liuanqi-libra7 commented 5 months ago

Hi @tangxiangyu0313 ,

The code in this link has already implemented quantization for YOLOv5. We provide two quantization modes for selection, and you can refer to the README for replication.

Thanks