Yolov5 从零开始目标检测训练及推理

Yolov5 从零开始目标检测训练及推理

环境搭建

硬件配置

任意 Intel 或 AMD 的 CPU
显存 不小于 4G 的 NVIDIA GPU

系统配置

Ubuntu 20.04

软件配置

GPU-Driver

参考博文

https://blog.csdn.net/qq_49466306/article/details/109705251

CUDA

参考博文

https://blog.csdn.net/qq_49466306/article/details/109727521

CUDNN

参考博文

https://blog.csdn.net/qq_49466306/article/details/109716798

Anaconda3

了解 Yolov5

尝试训练 Yolov5

参考博文进行最简单的配置和训练，工具以及安装方法均已列出，部分内容需要根据自己的路径或文件名称进行修改

https://blog.csdn.net/qq_49466306/article/details/119784211

尝试使用 Yolov5 进行推理

参考 Yolov5 官方说明中 Inference 部分，共有两种方式，直接加载训练好的模型以及图片进行推理，以及使用源码中的 detect.py 进行推理

https://github.com/ultralytics/yolov5

尽情享用吧~