Faster-RCNN.pytorch的搭建、使用过程个人记录(适配PyTorch 1.0以上版本)

Faster-RCNN.pytorch的搭建、使用过程个人记录

前言

本文主要记录(1)如何跑通源代码;(2)配置过程中遇到的错误调试过程。
在参考文章基础上添加了自己使用过程中遇到的问题。
参考文章:
Faster-RCNN.pytorch的搭建、使用过程详解(适配PyTorch 1.0以上版本)

faster-rcnn pytorch代码下载

我自己的配置环境:
python3.7.7+cuda10.1.243+Ubuntu16.04+Pytorch1.0。 英伟达显存8G

faster-rcnn pytorch配置过程

1.安装torch
请移步至https://pytorch.org/get-started/locally/查看相应的安装命令,安装最新版的pytorch包。参考执行代码如下:

conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch

2. 下载源码到本地
具体方法有两种:
(1)git clone到本地

git clone -b pytorch-1.0 --single-branch --depth=1 --recursive https://github.com/jwyang/faster-rcnn.pytorch.git

其中:-b是分支名,--single-branch是clone指定分支的命令,--depth==1是指克隆最近一次更改,--recursive用于循环递归克隆子项目。
(2)直接下载zip文件至本地
在这里插入图片描述
我使用的第二种。
3.创建数据文件夹
在本地创建一个文件夹专门用来搭建Faster-RCNN模型,所有的配置文件均在此目录下进行(如:“/detectron”)。

首先,将源码的zip文件copy至目标目录/detectron,同时解压缩得到如下所示文件夹;
在这里插入图片描述
其次,创建data文件夹,具体操作如下:

1. cd 进入project根目录
2. mkdir data   # 创建data文件夹

或者,

  • 1
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值