conv-neural-network - 如何在 MobileNet-SSD caffe 上进行增量学习

Question

我正在用20k图像训练我的分类器，但每周我都会得到更多新图片，所以我想逐步训练我以前的模型（最后停止的迭代），而不是再次在所有图像上重新训练20k+new_images，这是浪费时间和计算

我想出了用 Yolo 进行增量训练，但似乎找不到任何在这里实现的 MobileNet-SSD caffe https://github.com/chuanqi305/MobileNet-SSD

要了解有关我在说什么的更多信息，请参阅： 如何在 yolov3.weights 的基础上进行增量训练并回答此处提及的内容：

 darknet.exe partial cfg/yolov3.cfg yolov3.weights yolov3.conv.105 105 

Answer 1

您需要传递上一次迭代train.sh而不是 73000 次迭代。完成训练后，可以在快照文件夹中找到新的迭代

if ! test -f example/MobileNetSSD_train.prototxt ;then
echo "error: example/MobileNetSSD_train.prototxt does not exist."
echo "please use the gen_model.sh to generate your own model."
exit 1
fi
mkdir -p snapshot

#Initiate a new training
$CAFFE_ROOT/build/tools/caffe train -solver="solver_train.prototxt" \
-weights="mobilenet_iter_73000.caffemodel" \
-gpu 0