YOLOv3&pytorch在ubuntu实现

YOLOv3

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

配置环境

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

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

安装

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

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

  1. pycharm打开YOLOv3文件作为工程目录,配置编译器为刚搭建环境yolov3的python3.5
  2. pycharm中terminal中运行python detect.py --images imgs --det det ,根据报错使用pip命令安装各种模块,其中安装PIL模块中可能报错,就要换成使用pillow模块;
  3. 所有包安装完毕就可以运行了。
    YOLOv3&pytorch在ubuntu实现_第1张图片

你可能感兴趣的:(目标检测)