Improved yolov5 quantization

Xilinx / Vitis-AI

Vitis AI is Xilinx’s development stack for AI inference on Xilinx hardware platforms, including both edge devices and Alveo cards.

https://www.xilinx.com/ai

Apache License 2.0

1.5k stars 635 forks source link

Improved yolov5 quantization #1458

Open lroy-hh opened 2 months ago

lroy-hh commented 2 months ago

Based on Yolov5s, a 160*160 small target detection head was added. In the vitis-ai2.5 version, the quantization script is used as follows, and the following picture error appears: 量化脚本.docx

微信图片_20240902193223 Can anyone help me solve this problem?