该项目是通过神经网络训练检测人眼识别的完整项目,包含具体的训练和测试过程的代码。
opencv
dlib
caffe
dataset里面是训练模型用到的图片数据集
eye_model是已经训练好的模型
genData为生成训练数据文件
test里面包好测试代码
train里面包含神经网络训练的相关文件
python dlib_main.py(注:将caffe的路径改为你电脑本地的路径)
1、修改creat_hdf5.py里面对应的文件路径,生成训练所需的hdf5数据集
list_file = "val.txt"#txt文件
path_data = ""#当前项目的根目录
path_save = ""#生成的hdf5存储的文件夹路径
size_hdf5 = 1000#设置每个hdf5的数量大小
tag = 'val_'#生成hdf5文件前缀
python creat_hdf5.py
2、修改网络文件里面对应的路径
3、执行
sh train.sh