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’