王权富贵：使用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 
    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 
    from datasets.factory import get_imdb
  File "/home1/yul/yzq/faster-rcnn.pytorch-1.0/lib/datasets/factory.py", line 15, in 
    from datasets.coco import coco
  File "/home1/yul/yzq/faster-rcnn.pytorch-1.0/lib/datasets/coco.py", line 23, in 
    from pycocotools.coco import COCO
  File "/home1/yul/yzq/faster-rcnn.pytorch-1.0/lib/pycocotools/coco.py", line 60, in 
    from . import mask
  File "/home1/yul/yzq/faster-rcnn.pytorch-1.0/lib/pycocotools/mask.py", line 3, in 
    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 
    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版本的问题，修改如下，参考。