1. 数据集命名方式:
点击这里:
数据集的同一命名方式
2. 生成train.txt的命令:
find -name *.jpeg |grep train | cut -d/ -f3-4 > train.txt
3. 生成val.txt的命令:
find -name *.jpeg |grep val | cut -d/ -f3-4 >val.txt
4. 生成标签的命令:
sed -i '1,70s/.*/& 0/' train.txt
- 1-70行标为0
sed -i'71,141s/.*/& 1/' train.txt
- 71-141行标为1
- 对val.txt标签类似
- 一定要注意"& 1"中间只有一个空格,也就是图像名称和标签之间只能有一个空格,否则会报错:
生成lmdb数据成功如下所示:
5. 后续训练步骤:
参考这里第3步以及之后