yolo算法python代码_python通过yolo算法识别图片中的对象

1 场景

YOLO全称You Only Look Once，YOLO实现了自动驾驶汽车等前沿技术中使用的实时对象检测。

可以实现对图像中的对象进行识别和定位，如下效果：

1.png

这里不研究算法，只研究，在python环境下，如何使用yolo进行图像中物体的识别和定位。

2 官网

2.1 官方地址

YOLO使用相关网站如下：

github地址：

2.2 百度网盘

如下为作者在百度网盘中备份的相关文件(配置文件+演示文件)：

3 版本

python：3.6.3

4 依赖

(1)安装cv2依赖

pip install opencv-python

(2)安装numpy依赖

pip install numpy

4 代码

4.1 准备

(1)创建yolo配置文件

创建配置文件夹：cfg

将配置文件下载后，拷贝到cfg配置文件中：

(2)创建测试文件夹

测试文件夹：yoloSrc

输出文件夹：yoloRes

4.2 引入依赖

import cv2

import numpy as np

import os

import time

4.3 定义函数