YOLOv5:代码详解、制作自己的数据集训练再到项目部署

YOLOv5:代码详解、制作自己的数据集训练再到项目部署

  • 前言
  • 代码详解
    • 网络结构
    • 损失函数
  • 训练自己制作的数据集
    • 制作数据集
    • 训练自己的数据集
    • 代码调试
  • TensorRT项目部署
    • 官方文档介绍
    • 项目代码

前言

Yolov5的学习已经结束,这里做一个汇总,以下链接均为之前写下的博文。

代码详解

网络结构

请看博文:从YOLOv5源码yolo.py详细介绍Yolov5的网络结构

损失函数

请看博文:从YOLOv5源码loss.py详细介绍Yolov5的损失函数

训练自己制作的数据集

制作数据集

请看博文:制作YOLOv5数据集

训练自己的数据集

请看博文:使用YOLOv5训练自己的数据集模型

代码调试

请看博文:YOLOv5tagV5.0调试过程中的遇到的问题

TensorRT项目部署

官方文档介绍

请看博文:深度学习项目部署:使用TensorRT通过ONNX部署Pytorch项目

项目代码

完整代码:TensorRT部署yoloV5源码

你可能感兴趣的:(目标检测,pytorch,深度学习,python)