王权富贵:使用pytorch复现Fater-RCNN

环境配置:(以实现)

GTX 1080TI

CUDA 8

python 2.7

pytorch 0.4

使用master分支

训练自己的数据集时候修改 demo.py 中的 pascal_classes 类别 和 lib/datasets/pascal_voc.py 中的self._classes 类别。


环境配置:

GTX 1080TI

CUDA 8

python 2.7

pytorch 1.0

使用pytorch-1.0分支


真实拥有的环境配置:

GTX 1080TI

CUDA 8

python 2.7

pytorch 1.1

使用pytorch-1.0分支

1)是编译的时候遇到pyton2.7中第三方库访问权限问题,用以下授权代码解决, 参考。

sudo chown -R $USER /usr/local/lib/python2.7

2)接下来对于导入COCO包的错误,如下:

Traceback (most recent call last):
  File "trainval_net.py", line 27, in <module>
    from roi_data_layer.roidb import combined_roidb
  File "/home1/yul/yzq/faster-rcnn.pytorch-1.0/lib/roi_data_layer/roidb.py", line 9, in <module>
    from datasets.factory import get_imdb
  File "/home1/yul/yzq/faster-rcnn.pytorch-1.0/lib/datasets/factory.py", line 15, in <module>
    from datasets.coco import coco
  File "/home1/yul/yzq/faster-rcnn.pytorch-1.0/lib/datasets/coco.py", line 23, in <module>
    from pycocotools.coco import COCO
  File "/home1/yul/yzq/faster-rcnn.pytorch-1.0/lib/pycocotools/coco.py", line 60, in <module>
    from . import mask
  File "/home1/yul/yzq/faster-rcnn.pytorch-1.0/lib/pycocotools/mask.py", line 3, in <module>
    from . import _mask
ImportError: cannot import name _mask

可以使用如下方法解决,参考。

cd data
git clone https://github.com/pdollar/coco.git 
cd coco/PythonAPI
make

3)训练自己的数据集时候修改 demo.py 中的 pascal_classes 类别 和 lib/datasets/pascal_voc.py 中的self._classes 类别。

4)训练的时候出现,如下错误:

Traceback (most recent call last):
  File "trainval_net.py", line 318, in <module>
    im_data.data.resize_(data[0].size()).copy_(data[0])
RuntimeError: set_sizes_contiguous is not allowed on Tensor created from .data or .detach()

这是因为pytorch版本的问题,修改如下,参考。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值