YOLOv3&pytorch在ubuntu实现

watermelonoops

已于 2023-03-01 10:01:31 修改

阅读量1.6k

点赞数

分类专栏：目标检测文章标签： YOLOv3 pytorch ubuntu 目标检测工程实现

于 2019-11-15 10:13:12 首次发布

本文链接：https://blog.csdn.net/sinat_33896833/article/details/103078962

版权

6 篇文章 3 订阅

订阅专栏

YOLOv3

YOLOv3：you only look once，是经典的目标检测算法，今天我们在ubuntu操作系统利用pytorch实现YOLOv3

我们在使用pytorch版本YOLOv3前需要自行下载Anaconda：
我们需要配置环境中的三项内容：
Python 3.5
OpenCV
PyTorch 0.4

从https://github.com/ayooshkathuria/pytorch-yolo-v3使用git命令或者下载成压缩包
Control+Alt+T打开命令行；
使用conda info --envs命令查看当前conda创建的环境
使用conda create --name yolov3 python=3.5创建环境，指定Python版本为3.5；
在命令行中使用source activate yolov3命令激活刚才创建的YOLOv3环境，在环境中安装另外两项内容
conda install -c menpo opencv命令安装opencv
使用conda install pytorch=0.4.1 cuda90 -c pytorch安装pytorch，版本为0.4，使用其他版本运行时可能报错，安装过程缓慢可能需要切换不同的conda源尝试

需要事先下载YOLO训练好的权重包，在命令行中使用如下命令下载，如果速度慢可以将https://pjreddie.com/media/files/yolov3.weights复制到迅雷下载链接中去，将下载好的文件放到前面下载的YOLOv3文件夹中，和detect.py同目录

wget https://pjreddie.com/media/files/yolov3.weights

pycharm打开YOLOv3文件作为工程目录，配置编译器为刚搭建环境yolov3的python3.5
pycharm中terminal中运行python detect.py --images imgs --det det ，根据报错使用pip命令安装各种模块，其中安装PIL模块中可能报错，就要换成使用pillow模块；
所有包安装完毕就可以运行了。