首先感谢两位优秀的博主。
https://blog.csdn.net/The_heart_of_robort/article/details/85224232
https://blog.csdn.net/weixin_43380510/article/details/83004127
记录自己踩过的坑
1)准备工作
数据集 + clone代码
2)很遗憾未能找到已经训练好的模型,所以只能自己动手丰衣足食,好在实验室的硬件条件给力,在voc数据集上进行了训练,训练时长大概15小时。
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/fas