1 Star 2 Fork 0

Jin12180000/casia-surf-2019-codes

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
readme.txt 1.50 KB
一键复制 编辑 原始数据 按行查看 历史
zzzkk2009 提交于 2019-02-26 16:54 . 'first'
1、代码和数据集存放
1.1 获取数据集,即CASIA-SURF文件夹,里面包含phase1和phase2两个文件夹,进入phase1,解压train.zip和valid.zip文件,
得到Training文件夹,train_list.txt文件、Val文件夹、val_public_list.txt文件;
1.2 将本项目整个文件夹casia-surf-2019-codes拷贝到CASIA-SURF目录里面,保证和phase1同目录;
2、环境安装(ubuntu 16.04):
2.1 打开命令终端,cd到casia-surf-2019-codes目录,执行: pip install -r requirements.txt
3、数据集预处理:
3.1 打开命令终端,cd到casia-surf-2019-codes目录,依次执行:
3.1.1 python data_preprocess.py (生成val数据集的list文件)
3.1.2 python data_preprocess.py --train (生成train数据集的list)
3.1.3 cd data
3.1.4 python im2rec.py train_depth_all_112_29266.lst ../../phase1 (生成train数据集的.rec文件,用于mxnet训练)
3.1.5 python im2rec.py val_depth_all_112_9608.lst ../../phase1 (生成val数据集的.rec文件,用于mxnet验证)
4、模型训练
4.1 打开命令终端,cd到casia-surf-2019-codes目录,执行: python train_depth_shufflenet_v2.py
注:1) 本人使用的是包含4块GPU的服务器,配置为:11G Memory, GeForce GTX 1080,
服务器配置:CPU:Intel(R) Core(TM) i7-6850K CPU @ 3.60GHz 12核,64G内存,1TB固态硬盘;
2) 训练过程使用了1,2,3块GPU进行训练,batch_size=2048;
5、测试
5.1 打开命令终端,cd到casia-surf-2019-codes目录,执行: python commit.py ../phase1/val_public_list.txt --load-epoch 554
注:程序执行完后,会在当前目录下生成 commit_depth_%Y-%m-%d.txt 格式的结果文件
commit.py 第一个参数为待提交成绩的list文件,第二个参数load-epoch表示:加载第load-epoch数的模型进行预测。
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/jin1218/casia-surf-2019-codes.git
git@gitee.com:jin1218/casia-surf-2019-codes.git
jin1218
casia-surf-2019-codes
casia-surf-2019-codes
master

搜索帮助