> 文章列表 > 目标检测框架yolov5环境搭建

目标检测框架yolov5环境搭建

目标检测框架yolov5环境搭建

    目前,目标检测框架中,yolov5 是很火的,它基于pytorch框架,集成opencv等框架,项目地址:https://github.com/ultralytics/yolov5,对我来说,机器学习、深度学习才开始接触,本篇只能是按照项目说明来进行一个简单的测试和验证。

    这里目标检测,其实就是给定一些图片后者视频,结合现有的模型训练,然后生成识别的结果。

    项目介绍给出的两个示例图片如下:

    第一张像是西班牙足球运动员皮克,夏奇拉的前夫弟弟:

    

    还有一个是齐祖打领带咆哮的照片:

 

    这里目标检测,需要识别图片中的人和车,还有领带。

    环境要求:python3.9及以上版本。

    我这里是在linux下,python版本是3.10 。

    1 、克隆github上的项目到本地:https://github.com/ultralytics/yolov5,如果这个地址比较慢,可以考虑国内的码云地址:https://gitee.com/monkeycc/yolov5,这个地址就是同步的github仓库的项目,但是下载速度很快:

git clone https://github.com/ultralytics/yolov5

    2、安装依赖 

cd yolov5
pip install -r requirements.txt

    3、下载模型文件:https://github.com/ultralytics/yolov5/releases/download/v7.0/yolov5s.pt

如果这个地址太慢,可以到这个百度网盘的地址上下载:https://pan.baidu.com/s/1vSqeX7WDyo23FCLOwiTBjA?pwd=yolo 

    下载之后,放到项目根路径下,我们暂时只下载一个模型文件yolov5s.pt。

   4、运行测试demo。

buejee@SKY-20230223WWS:~/yolov5$ python3 detect.py
detect: weights=yolov5s.pt, source=data/images, data=data/coco128.yaml, imgsz=[6                                         40, 640], conf_thres=0.25, iou_thres=0.45, max_det=1000, device=, view_img=False                                         , save_txt=False, save_conf=False, save_crop=False, nosave=False, classes=None,                                          agnostic_nms=False, augment=False, visualize=False, update=False, project=runs/d                                         etect, name=exp, exist_ok=False, line_thickness=3, hide_labels=False, hide_conf=                                         False, half=False, dnn=False, vid_stride=1
YOLOv5 🚀 2023-4-10 Python-3.10.6 torch-2.0.0+cu117 CPUFusing layers...
YOLOv5s summary: 213 layers, 7225885 parameters, 0 gradients
image 1/2 /home/buejee/yolov5/data/images/bus.jpg: 640x480 4 persons, 1 bus, 197                                         .4ms
image 2/2 /home/buejee/yolov5/data/images/zidane.jpg: 384x640 2 persons, 2 ties,                                          139.0ms
Speed: 4.6ms pre-process, 168.2ms inference, 9.4ms NMS per image at shape (1, 3,                                          640, 640)
Results saved to runs/detect/exp2
buejee@SKY-20230223WWS:~/yolov5$

     根据提示,结果放到了runs/detect/exp2目录下。

    先看看大致的结果:

    图片中的人都识别出来了,并且用红框画出来了。

    具体的图片:

    第一张,公交车与旁边的人:

 

    第二张,人和领带都识别了:

 

   感觉很强大,很厉害,环境部署也不是很复杂,基本就是傻瓜式的执行几个命令。