基于YOLOv5的王镇短袖短裤识别技术介绍

本文主要介绍基于YOLOv5的王镇短袖短裤识别技术。针对检测分类的精度和模型运行的速度进行权衡，采用了检测精度较好的One-stage检测框架，并结合模型压缩和加速方法，实现了模型的快速运行。在数据增广和训练预训练模型等方面也做了详细的介绍，同时对输入图像尺寸的优化选择也进行了探讨。最终实验结果表明，使用TensorRT和Slimming后的8分类模型在王镇短袖短裤的识别上具有较高的准确率和速度优势。