训练集与label转换成tfrecord文件时乱序
训练大量图片数据集时将图片转换成tfrecord文件可以提高训练效率,但是在图片集在转换过程中,当图片文件夹名称以数字排列顺序时,在转换过程中如果只是使用sorted处理时,排序的结果并不是简单的从小到大,因为名称时string可是 所以排序结果为
class_folders = sorted(class_folders)
此时为了得到正确的排序,需要将string格式转换成int类型,然后在使...
原创
2019-08-21 15:59:39 ·
332 阅读 ·
0 评论