classify 文件夹包含用于训练、验证和部署 YOLOv5 分类模型的脚本。自 2022 年 8 月起,YOLOv5 支持分类任务。

可以使用 classify/train.py 脚本在自定义数据集上训练 YOLOv5 分类模型,使用 classify/val.py 脚本验证模型的准确性,并使用 classify/predict.py 脚本在图像、视频或实时摄像头流上运行推理。

<aside> 💡 主要用于训练分类任务的分类模型

</aside>