环境配置:(以实现)
GTX 1080TI
CUDA 8
python 2.7
pytorch 0.4
训练自己的数据集时候修改 demo.py 中的 pascal_classes 类别 和 lib/datasets/pascal_voc.py 中的self._classes 类别。
环境配置:
GTX 1080TI
CUDA 8
python 2.7
pytorch 1.0
真实拥有的环境配置:
GTX 1080TI
CUDA 8
python 2.7
pytorch 1.1
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版本的问题,修改如下,参考。