图像视频处理领域,目标检测是很常见的一种应用,而其中比较经典的方法是yolo,而且目前已经出现了好几个版本,笔者用过v3,目前已经用到了V5版本。
YoloV5版本的前提要求为:Python>=3.7.0 并且 Pytorch>=1.7
第一步 下载安装
git clone https://github.com/ultralytics/yolov5 # clone
cd yolov5
pip install -r requirements.txt # install
第二步 直接命令行测试
python detect.py --source img.jpg
即可对img.jpg图像进行目标检测,检测结果会存放在 run/exp的相关目录下。
以上这是对于Yolov5的最简单应用,之后笔者会介绍一下如果自定义图像数据集进行目标识别。