CNN中的图像分类

时间:2020-05-17 13:42:58

标签: tensorflow keras deep-learning classification conv-neural-network

我有一个图像数据库和每个图像的Category的excel表。如何将两者合并以将其用于训练过程? 例如:在excel工作表中:给出了图像名称和类别

2 个答案:

答案 0 :(得分:0)

您有多少张图片?如果不是太多,可以将它们全部加载到内存中。保留图像列表和标签列表。

您可以通过遍历excel工作表来做到这一点,对于每一行,在标签列表中插入标签,然后将相应的图像加载到图像列表中。

然后,您可以将这两个列表提供给Keras的模型。fit

希望有帮助。

答案 1 :(得分:0)

我想的问题是,您的图像在一个数据集中,而它们相应的标签在Excel文件中,对吗?您想知道如何将它们组合并输入模型。好吧,如果问题是这样,则可以将图像作为x_train / x_test / x_val加载,并判断类别为y_train / y_test / y_val。然后将它们作为输入输入模型。 (当然,您需要按image_name映射它们)

或更直观的方法是,您可以使用熊猫中的read_csv文件将excel加载为数据框,并将所有图像名称另存为另一个数据框。接下来,将两个数据框合并为一个,以图像名称为键。然后,以下过程在每次模型训练中都是相同的。希望对您有所帮助。