run_faster rcnn

ubuntu

训练技巧

• 使用固定cuda

CUDA_VISIBLE_DEVICES=1 python trainval_net.py --cuda --r true --checksession 1 --checkepoch 65 --checkpoint 455

• 使用nohub将程序在xshell挂起
  nohup—将程序以忽略挂起信号的方式运行起来
  nohup - 将程序以忽略挂起信号的方式运行起来

nohup python demo.py &

查看运行情况

tail -f nohup.out

• demo.py 检测时输出到固定文件夹/预测xml结果保存
  Faster-RCNN.pytorch的搭建、使用过程详解（适配PyTorch 1.0以上版本）
  

• 画图
  使用pytorch版faster-rcnn训练自己数据集
  评论区第6页
  
  
  
  修改300检测框数量
  
  xml属性修改
  
  使用mmdetect

• 画PR曲线，输出rec,prec
  基于Faster RCNN，如何画出类别的Precision-recall曲线？

• 编译问题
  faster-rcnn实现目标检测
  
  
  

参考的博客

github

注释Yang Jianwei 的Faster R-CNN代码（PyTorch）

使用faster-rcnn.pytorch训练自己数据集（完整版）

–可看这个博客的其他参考

注意选择版本未pytorch1.0版本
faster-rcnn.pytorch-1.0的jwyang当前最火版本代码复现与讲解

faster-rcnn实现目标检测

使用pytorch版faster-rcnn训练自己数据集

安装包

conda create -n torch1.0 python=3.6

pip install torch===1.0.1 torchvision===0.2.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple

pip install -r requirements.txt
pip install imageio

路径和环境配置

faster rcnn训练自己的数据集（pytorch版）

错误

• List item

TypeError: load() missing 1 required positional argument: ‘Loader‘

d1=yaml.load(file,Loader=yaml.FullLoader)
d1=yaml.safe_load(file)
d1 = yaml.load(file, Loader=yaml.CLoader)

• List item
  ImportError: cannot import name ‘_mask’
  解决：

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

• List item

RuntimeError: Not implemented on the CPU (ROIAlign_backward at /faster-rcnn.pytorch/lib/model/csrc/ROIAlign.h:44)
解决：

python trainval_net.py --cuda

• List item
  添加链接描述IndexError: list index out of range

解决：删除fast-rcnn-master/data/cache/ 文件夹下的.pkl文件，或者改名备份，重新训练即可。

• List item
• faster-rcnn.pytorch问题汇总和解决
  
• List item
  训练为nan
  出现训练结果为nan,看官网的issues,是标签是图像边缘，删除或修改这些标签
• List item

训练提示 无weight
删除存放数据集集中生成的cache

resume

faster_rcnn_1_6_465
6–checkepoch
465–checkpoint
load_name = os.path.join(output_dir,
‘faster_rcnn_{}{}{}.pth’.format(args.checksession, args.checkepoch, args.checkpoint))

windows make编译出错

use:

Faster RCNN pytorch 1.0版调试（踩坑）过程记录

make: *** [all] Error 2 when building Trilinos

其他参考

记pytorch版faster rcnn配置运行中的一些坑

运行pytorch版faster-rcnn遇到的问题

解决 : ModuleNotFoundError: No module named ‘pycocotools._mask’