老师,我想用自己的数据集来进行训练实现图像分类,数据集格式应该是什么样子的?以及如何标注,如何加载数据?
来源:4-6 Inception-mobile_net(1)
炼丹师
2018-09-15
写回答
1回答
-
正十七
2018-09-19
加载数据你可以参考咱们的cifar10的数据产生器,这个在实战中我详细讲过。
如果你是用自己的数据,需要你读入图片,将图片转换成numpy矩阵。lable的话是你自己去设置的,比如你要把数据分为10个类,那么你就把图像类别值设为[0, 1,2,3,4,5,6,7,8,9]10个数中的一个,然后这个标注工作是需要人力去做的。
做完上述工作后,你就得到了一个四维矩阵来存储所有图像,和一个向量来表示每张图像的类别。比如矩阵的大小可能是[10, 128, 128, 3], 向量的大小为(10,),这样,你就有10张128×128的3通道图像,矩阵中的图像和向量中的类别是一一对应的。
有了这两个东西之后,其他的就可以参考咱们的cifar10的数据产生器了。
10
相似问题