码迷,mamicode.com
首页 > 其他好文 > 详细

基于YOLO-V2的行人检测(自训练)附pytorch安装方法

时间:2020-05-03 14:36:33      阅读:163      评论:0      收藏:0      [点我收藏+]

标签:关于   mod   下载链接   文件中   user   安装   move   http   dex   

声明:本文是别人发表在github上的项目,并非个人原创,因为那个项目直接下载后出现了一些版本不兼容的问题,故写此文帮助解决。(本人争取在今年有空的时间,自己实现基于YOLO-V4的行人检测)

项目链接:https://github.com/emedinac/Pedestrain_Yolov2

此项目是基于pytorch框架的实现方案。

本文介绍一下ubuntu安装pytorch的安装方法

直接采用pip install pytorch的下载实在太感人,因此先更改一下pip源。

本人采用的是阿里源,实现方法:

1. 在~目录下新建.pip文件夹。

mkdir ~/.pip

2.进入.pip文件夹

cd ~/.pip

3.新建文件pip.conf

gedit pip.conf

输入一下内容并保存退出

[global]
index-url = http://mirrors.aliyun.com/pypi/simple/
[install]
trusted-host=mirrors.aliyun.com

4.安装pytorch

pip install torch torchvision

下载速度不是一般地快。

接下来就是训练网络,在执行这一句

python train.py cfg/person.data cfg/yolo_person.cfg backup/yolo_person.weights
时出现
volatile was removed and now has no effect. Use `with torch.no_grad():`
修改train.py文件中的 output = model(data).data
在这一句的前面加上
with torch.no_grad():
  output = model(data).data
以及在output = model(data)这一句的前面加上下面这一句。并缩进
with torch.no_grad():
  output = model(data)
其余的都不需要更改缩进。


关于下载速度的问题,数据集给出的下载链接都是国外的网站,下载速度极慢,推荐一个加速器。--蚂蚁加速器
下载链接:https://cov.lausera.com/aff/ahxwF
然后就是手机打开这个加速器,一般选用那个日本或者新加坡的,再开热点给电脑,下载速度可以达到1m/s。

 

基于YOLO-V2的行人检测(自训练)附pytorch安装方法

标签:关于   mod   下载链接   文件中   user   安装   move   http   dex   

原文地址:https://www.cnblogs.com/wxkang/p/12821566.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!