faster-rcnn.pytorch 学习

首先感谢两位优秀的博主。

https://www.geek-share.com/detail/2756556201.html

https://www.geek-share.com/detail/2750189012.html

记录自己踩过的坑

1）准备工作

数据集 + clone代码

2）很遗憾未能找到已经训练好的模型，所以只能自己动手丰衣足食，好在实验室的硬件条件给力，在voc数据集上进行了训练，训练时长大概15小时。

[code]CUDA_VISIBLE_DEVICES=1 python trainval_net.py --dataset pascal_voc --net vgg16 --bs 1 --nw 4  --cuda --load_dir /home/whut/yyCode/faster-rcnn.pytorch/data/models

CUDA_VISIBLE_DEVICES：gpu的id

-–dataset：数据集，eg：pascal-voc。

-–net：backbone网络，eg：vgg16。

–bs：batch size

–nw：worker number,取决于Gpu能力

-–cuda：使用gpu

每完成一个epoch则会在/home/whut/yyCode/faster-rcnn.pytorch/models/vgg16/pascal_voc下生成一个已经训练好的，可用作检测的model。

vggnet16

 resnet50

3）测试

[code]python test_net.py --dataset pascal_voc --net vgg16 --checksession 1 --checkepoch 17 --checkpoint 10021  --cuda

训练好的模型：faster_rcnn_1_17_10021

checksession = 1，checkepoch = 17， checkpoint = 10021

4）demo命令

[code]python demo.py --net vgg16 --checksession 1 --checkepoch 17 --checkpoint 10021 --cuda --load_dir /home/whut/yyCode/faster-rcnn.pytorch/models