本文不涉及对Yolov3的理解,仅讲解如何跑通程序。
针对开源项目PyTorch-YOLOv3,适合新手看
环境:Win10+CUDA10.0+python3.6(后来因为另一个项目重装了CUDA10.2也能跑)
原来的实现是在Linux环境,搬到win10有一些问题。不过问题不大。
下面按照作者的说明顺序来讲解。过程中需要下载的大文件我放在了百度网盘,需要的同学可以自取。提取码在文末。
安装
clone and install requirements
git clone https://github.com/eriklindernoren/PyTorch-YOLOv3
首先很多人win10的命令行窗口没有git指令,两种方法,要么直接download下来,要么自己装一个git for windows,建议后者,因为同时还能下载到bash等指令,都会常用的。
cd PyTorch-YOLOv3/
pip install -r requirements.txt
win10下不需要sudo,也不用pip3。
安装requirements里的东西时可能因为网络波动一片红,出现那种情况就一步一步的pip install -i https://pypi.tuna.tsinghua.edu.cn/simple xxx(使用国内源)。每个库的版本,还有点说法的,主要是TensorFlow的版本影响比较大(1.x和2.x版本差的挺多)。最终的环境配置是这样的。