【问题标题】:How to do incremental learning on MobileNet-SSD caffe如何在 MobileNet-SSD caffe 上进行增量学习
【发布时间】:2019-12-27 09:03:43
【问题描述】:

我在 20k 图像上训练我的分类器,但每周我都会得到更多新图片,所以我想增量训练我以前的模型(最后停止的迭代)而不是重新训练 20k+new_images 再次处理所有图像,这是浪费时间和计算

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

要了解更多关于我所说的内容,请参阅: How to do incremental training on the basis of yolov3.weights & 回答此提及here

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

【问题讨论】:

    标签: conv-neural-network caffe mobilenet


    【解决方案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
    

    【讨论】:

    • 快速提问。当我像这样在 caffe 中进行增量学习时,我使用什么数据集? (在这个例子中,之前的 20k + 新数据,或者只是新数据)
    猜你喜欢
    • 2020-08-06
    • 2023-01-18
    • 2018-11-07
    • 1970-01-01
    • 2021-02-24
    • 2017-11-08
    • 2019-03-06
    • 2016-01-10
    • 1970-01-01
    相关资源
    最近更新 更多