老师,我想用自己的数据集来进行训练实现图像分类,数据集格式应该是什么样子的?以及如何标注,如何加载数据?

来源: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的数据产生器了。

1
0

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学,深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

查看课程