python - 用于计算机视觉的 Python 中的 LDA

Question

我正在使用 LDA（潜在狄利克雷分配）主题建模进行基于内容的图像检索。我希望使用 Python 来做同样的事情。我无法在 Python 中为图像数据集的 LDA 找到任何库/包。这个包似乎只适用于文本语料库。请为图像语料库推荐任何 Python 包，或者简要列出执行图像 LDA 所需的步骤。

Answer 1

我最近在 python 中基于视觉词使用 LDA 实现了图像检索。查看我的github页面了解详细信息。

视觉词是在图像关键点处提取的图像特征的聚类中心。在这个例子中，密集的 SIFT 特征是从 40 个不同人的 10 张人脸图像（Olivetti 人脸数据集）的集合中提取的。

使用 mini-batch K-means 将 128 维 SIFT 特征聚类到 K 个视觉词的字典中。在线变分贝叶斯算法用于学习 LDA 主题模型并提取主题比例以用于训练图像数据。

然后将测试图像数据转换为主题空间，并根据训练图像和测试图像的主题比例之间的余弦相似度检索训练图像。