CNN训练自己的数据

1、准备数据

将自己的数据放到train  和 val 文件夹下，然后使用 create_imagefile.py 制作txt列表文件,生成train.txt和val.txt

2、使用 shell 脚本 create_mydata.sh 制作lmdb格式数据

3、通过脚本make_mydata_mean.sh制作均值文件mydata.binaryproto

4、修改网络配置proto文件和求解文件

note:

1、在用python代码读取数据时，如果出现 这个错误

gentech@gentech-Precision-T3600:~/source_tools/caffe/data/mydata$ python read_lmdb.py Traceback (most recent call last):
File "read_lmdb.py", line 7, in <module>
import lmdb
ImportError: No module named lmdb

那么用 pip install lmdb