本文详细介绍了在Ubuntu系统下训练图像数据集的四个步骤：数据准备，图片转换为lmdb格式，计算均值，以及开始训练。强调了在每个步骤中需要注意的细节，如txt文件的正确性，图片尺寸的一致性，以及prototxt文件的修改等。
第一步:准备数据
首先需要准备两个文件夹,一个是train,一个是val.收集一定数量的图片,越多越好,要求同样的分辨率和扩展名.
train中,将图片分类存放至文件夹中,例如,000文件夹中全是赵XX ,001文件夹中全是钱XX,如下
val中,可以不按照文件夹存放,也可以如train一样按照文件夹存放,同样存放一定数量的图片,例如赵XX和钱XX的图片
接下来就是准备对应的索引文件train.txt和val.txt,里面包括:图片相对于txt文件的相对位置和编号,train.txt和val.txt格式一致,如下
其中,图片的对应关系一定不能错,train.txt中赵XX 是2,则val.txt中赵XX 的图片对应也必须是2.
第二步,图片转换
准备好图片后,就需要将图片转换成lmdb格式,需要修改examples/imagenet中的create_imagenet.sh.该文件如图所示
最低0.47元/天 解锁文章
308
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
