目标检测 |
您所在的位置:网站首页 › 复现算法 › 目标检测 |
本文不涉及对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等指令,都会常用的。 win10下不需要sudo,也不用pip3。 安装requirements里的东西时可能因为网络波动一片红,出现那种情况就一步一步的pip install -i https://pypi.tuna.tsinghua.edu.cn/simple xxx(使用国内源)。每个库的版本,还有点说法的,主要是TensorFlow的版本影响比较大(1.x和2.x版本差的挺多)。最终的环境配置是这样的。 预训练权重的下载,有了它就可以直接检测啦(只能检测数据集中有的类别)。 cd weights/ bash download_weights.sh这里的bash指令就体现出刚刚装git的方便性了。不过看一下.sh文件中的内容,三句wget -c ...的下载指令,这个win10也是没有的,下载安装一下GNU wget,按照方法一步步来就行。 装好了之后,可以继续执行bash指令,也可以把那三句wget依次执行。下载好权重得到的是这样的文件结构 数据集的下载,使用的COCO数据集是常用于目标分类的数据集。 cd data/ bash get_coco_dataset.sh和刚刚的过程类似,看一下内部的代码。可以直接执行指令,如果中途出错,可能是网络波动导致下载失败或者win10没有这个指令等。新手建议按下面的讲解一步步来,也方便理解。 # Clone COCO API git clone https://github.com/pdollar/coco cd coco mkdir images cd images # Download Images wget -c https://pjreddie.com/media/files/train2014.zip wget -c https://pjreddie.com/media/files/val2014.zip # Unzip unzip -q train2014.zip unzip -q val2014.zip cd .. # Download COCO Metadata wget -c https://pjreddie.com/media/files/instances_train-val2014.zip wget -c https://pjreddie.com/media/files/coco/5k.part wget -c https://pjreddie.com/media/files/coco/trainvalno5k.part wget -c https://pjreddie.com/media/files/coco/labels.tgz tar xzf labels.tgz unzip -q instances_train-val2014.zip # Set Up Image Lists paste |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |