一、 制作lmdb数据源
先建立一个文件夹,文件夹内包含train、val。其中文件夹train下的数据作为训练数据的初始数据。Val下的数据作为测试时的初始数据。
编写lmdb.sh文件如下:
Train.txt文件的内容如下:
Val.txt文件的内容如下:
在终端中输入sh lmdb.sh即可开始制作lmdb数据源
二、编写train.sh文件,具体内容如下:
三、编写solver.proto.txt文件,具体内容如下(注意文件中的路径最好为绝对路径):
四、以下为所有需要的文件
Detect_train_lmdb、detect_val_lmdb为已经制作好的lmdb数据源。
Model文件夹为训练过程中快照下来的数据。Train_test.prototxt为之前编写好的网络配置文件。
以上操作全部执行完毕后就可以在当前路径下训练网络了,在终端中输入sh train.sh即可训练。