Faster RCNN学习笔记

• 学习的链接https://github.com/jwyang/faster-rcnn.pytorch
• 安装pytorch：
  • 安装GPU版本Pytorch
  • 链接中要求安装0.4.0版本，但实际会遇到AttributeError: 'module' object has no attribute 'ModuleDict'问题，需要安装0.4.1版本解决该问题（pip install torch==0.4.1）
• 在pip install -r requirements.txt过程中遇到下列问题：
  • 如果采用python3.6版本，会出现No matching distribution found for skbuild，这时手动 pip install scikit-build即可解决
  • 如果采用python2.7版本，会出现Command "python setup.py egg_info" failed with error code 1 in /tmp/pip-build-L4eqAl/opencv-python/， 这时修改requirements.txt中opencv-python为opencv-python==3.1.0.0即可
• 执行训练命令CUDA_VISIBLE_DEVICES=$GPU_ID python trainval_net.py --dataset pascal_voc --net vgg16 --bs $BATCH_SIZE --nw $WORKER_NUMBER --lr $LEARNING_RATE --lr_decay_step $DECAY_STEP --cuda
  • imdb模块找不到情况，这时export PYTHONPATH=~/faster-rcnn.pytorch-master/lib/datasets/即可
  • 存在No module named _tkinter, please install the python-tk package问题，采用sudo apt-get install python-tk，注意不要采用pip安装模式。

  • torch.jit.frontend.NotSupportedError: slicing multiple dimensions at the same..问题，采用在 @torch.jit.script 这个注释前加注释 ：# @torch.jit.script解决

•