为机器人换个好使的脑子(十六)

现在,我们可以开始训练卷积神经网络了。我们用的是MobileNet-SSD。

  1. 下载MobileNet-SSD
$ cd $CAFFE_ROOT/examples
$ git clone --depth 1 https://github.com/chuanqi305/MobileNet-SSD
  1. 创建lmdb数据链接
$ cd MobileNet-SSD
$ ln -s $HOME/Sean/Sean/lmdb/Sean_test_lmdb test_lmdb
$ ln -s $HOME/Sean/Sean/lmdb/Sean_train_lmdb train_lmdb
$ ln -s $HOME/Sean/labelmap.prototxt labelmap.prototxt
  1. 生成train/test/deploy prototxt
$ ./gen_model.sh 5    //5为类别数,要加上background
在example下生成3个prototxt:
  1. 训练
编辑$CAFFE_ROOT/examples/MobileNet-SSD/train.sh
#!/bin/sh
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
../../build/tools/caffe train -solver="solver_train.prototxt" \
-weights="mobilenet_iter_73000.caffemodel" -gpu=0
开始训练
$ ./train.sh