1.初次训练命令行:
python ./tools/train_net.py --gpu 0 --solver ./models/pascal_voc/VGG_CNN_M_1024/faster_rcnn_end2end/solver.prototxt --weights ./data/imagenet_models/VGG_CNN_M_1024.v2.caffemodel --imdb voc_2007_trainval --iters 90000 --cfg ./experiments/cfgs/faster_rcnn_end2end.yml
在solver.prototxt文件中修改
snapshot: 10000
2.Faster-rcnn使用snapshot继续训练
请参考博客:R-FCN/Faster-rcnn使用snapshot继续训练_Hytz的博客-CSDN博客
修改完代码以后继续训练:
python ./tools/train_net.py --gpu 0 --solver ./models/pascal_voc/VGG_CNN_M_1024/faster_rcnn_end2end/solver.prototxt --snapshot vgg_1024_faster_rcnn_iter_60000.solverstate --imdb voc_2007_trainval --iters 90000 --cfg ./experiments/cfgs/faster_rcnn_end2end.yml