课程链接:https://edu.51cto.com/course/31295.html
PyTorch版的YOLOv5是一个当前非常流行的目标检测器,本课程讲述对YOLOv5进行更换骨干网(backbone)的方法,使其更加轻量或提高性能。
本课程在YOLOv5 v6.1版本代码的基础上更换其骨干网,在Windows和Ubuntu系统上演示针对自己的数据集进行骨干网更换、重头训练和性能评估过程,并讲解骨干网网络模型原理以及原代码针对骨干网更换的修改部分。
课程中使用和更换了如下的骨干网:YOLOv5s、YOLOv5n、MobileNetV2、MobileNetV3、 ShuffleNetV2、GhostNet、Transformer
项目实战过程包括:PyTorch环境安装、YOLOv5项目安装、准备自己的数据集、修改配置文件、训练网络模型以及性能评估。