目标检测

您所在的位置:网站首页 复现算法 目标检测

目标检测

2023-08-17 19:28| 来源: 网络整理| 查看: 265

本文不涉及对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版本差的挺多)。最终的环境配置是这样的。 在这里插入图片描述在这里插入图片描述在这里插入图片描述 框起来的是版本比较重要的,当然并不是说其他的版本不可以。实际上并不是所有的库都用到了,大家在配环境的时候就按着教程来,不需要把我列出来的都一一对应。

Download pretrained weights

预训练权重的下载,有了它就可以直接检测啦(只能检测数据集中有的类别)。

cd weights/ bash download_weights.sh

这里的bash指令就体现出刚刚装git的方便性了。不过看一下.sh文件中的内容,三句wget -c ...的下载指令,这个win10也是没有的,下载安装一下GNU wget,按照方法一步步来就行。 在这里插入图片描述

装好了之后,可以继续执行bash指令,也可以把那三句wget依次执行。下载好权重得到的是这样的文件结构 在这里插入图片描述

Download COCO

数据集的下载,使用的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