YOLOv5改进:更换骨干网(Backbone)

课程链接:https://download.csdn.net/course/detail/37465

PyTorch版的YOLOv5是一个当前非常流行的目标检测器,本课程讲述对YOLOv5进行更换骨干网(backbone)的方法,使其更加轻量或提高性能。

本课程在YOLOv5 v6.1版本代码的基础上更换其骨干网,在Windows和Ubuntu系统上演示针对自己的数据集进行骨干网更换、重头训练和性能评估过程,并讲解骨干网网络模型原理以及原代码针对骨干网更换的修改部分。

课程中使用和更换了如下的骨干网:YOLOv5s、YOLOv5n、MobileNetV2、MobileNetV3、 ShuffleNetV2、GhostNet、Transformer

项目实战过程包括:PyTorch环境安装、YOLOv5项目安装、准备自己的数据集、修改配置文件、训练网络模型以及性能评估。

YOLOv5改进:更换骨干网(Backbone)_第1张图片

 

你可能感兴趣的:(深度学习之计算机视觉,python,深度学习,pytorch,人工智能)