如何使用未标记的图像训练 CNN?

Posted

技术标签:

【中文标题】如何使用未标记的图像训练 CNN?【英文标题】:How do I train CNN using unlabelled images? 【发布时间】:2018-03-01 01:47:12 【问题描述】:

我正在尝试使用 CNN GoogleNet Inception 训练图像分类器。我有一些标记的图像(每个类别 1000 个)和更多未标记的图像。到目前为止,我只使用了标记的图像并且我得到了很好的准确性。我只是不确定是否可以使用某种未标记的图像。

关于它们的唯一信息是,一个目录中总是有一些图像(1-10)。并且一个目录下的图片属于同一类。

谢谢

【问题讨论】:

【参考方案1】:

看看Keras ImageDataGenerator。它是一个方便的函数,可以从对应于类的子目录中读取图像。

即使您不使用 Keras 进行训练,您也可以进行一次虚拟运行来为未标记的图像生成标签,然后在您的神经网络架构中使用这些标签。

您还可以查看 pseudo labelling 以获取您没有任何有关内容信息的图像。

【讨论】:

以上是关于如何使用未标记的图像训练 CNN?的主要内容,如果未能解决你的问题,请参考以下文章

如何使用训练有素的 Keras CNN 模型对新的未标记数据进行预测

如何使用经过训练的 CNN 模型(torch7 格式)进行单张图像推理?

如何在训练 CNN 期间删除重复项?

MATLAB深度学习采用Faster R-CNN实现车辆目标检测

MATLAB深度学习采用Faster R-CNN实现车辆目标检测

如何在 Keras 中使用 predict_generator 对未标记的测试数据执行预测?