YOLOv5视觉AI库安装

打开YOLOv5开源仓库:

https://github.com/ultralytics/yolov5/blob/master/README.zh-CN.md

下载源码:

YOLOv5视觉AI库安装_第1张图片

安装 : pip install -r requirements.txt

YOLOv5视觉AI库安装_第2张图片

完成安装

YOLOv5视觉AI库安装_第3张图片

目标检测推理

YOLOv5视觉AI库安装_第4张图片

可通过PyTorch Hub加载YOLOv5检测模型

YOLOv5视觉AI库安装_第5张图片

检测图像并返回数据帧

使用YOLOv5要先安装opencv-python和pandas库

安装opencv-python

YOLOv5视觉AI库安装_第6张图片

安装pandas

YOLOv5视觉AI库安装_第7张图片

安装IPython

YOLOv5视觉AI库安装_第8张图片

安装PIL: pip install pillow

YOLOv5视觉AI库安装_第9张图片

安装相关依赖:

pip install seaborn

pip install matplotlib

pip install tqdm

pip install PyYaml

pip install torchvision

pip install psutil

成功打印检测结果

YOLOv5视觉AI库安装_第10张图片

你可能感兴趣的:(Python,人工智能,计算机视觉)