我有一组图像数据,想要根据内容将它们分成不同的组。目前为止,我尝试找到图像的中位数,并尝试基于中位数值将它们分成不同的聚类。但是,我该如何做才能将我的图像聚类分组呢?我在谷歌上搜索了很多聚类相关的内容,但是大多数结果都是基于颜色而不是图像内容进行聚类。请问是否有人可以提供我有关这方面信息的答案?我是否可以自动地使用中位数或其他技术来对数据集进行聚类分组呢?
from PIL import Image
import numpy as np
import os
Median=[]
k=[]
def get_imlist(path):
return [os.path.join(path,f) for f in os.listdir(path) if f.endswith('.jpg')]
path='D:/Images/dataset'
imlist= get_imlist(path)
for file in imlist:
head,tail=os.path.split(file)
im=np.array(Image.open(file).convert('L'))
m=np.median(im)
M=[m,tail]
print '.'
Median.append(M)
Results=sorted(Median, key=lambda median: median[0])
print Results