个回答

• louisgry

  2017-06-09 00:24:09

  import colorsys
  import os
  import shutil
  import numpy as np
  from PIL import Image
  import kmeans
  
  class ImagesCluster(object):
      def __init__(self, imagedir, k):
          self._imageDir = imagedir
          self._image2VectorDir = os.path.join(imagedir, 'image2vector')
          if not os.path.isdir(self._image2VectorDir):
              os.mkdir(self._image2VectorDir)
          self._imageVectorFile = os.path.join(self._image2VectorDir, 'images.txt')
          self._k = k
          for i in range(self._k):
              clusterDir = os.path.join(self._imageDir, 'cluster --{}'.format(i))
              if os.path.isdir(clusterDir):
                  os.mkdir(clusterDir)
  
      def _loadImages(self):
          images = os.listdir(self._imageDir)
          imagesfile = [os.path.join(self._imageDir, image) for image in images]
          return imagesfile
  
      def _hsv2L(self, h, s, v):
          QH =0
          if (h<=20) or (h>=315):
              QH =0
          if (h>20 and h<=40):
              QH=1
          if (h>40 and h<=75):
              QH =2
          if (h>75 and  h<=155):
              QH =3
          if (h>155 and h<=190):
              QH = 4
          if (h>190 and  h<=270):
              QH = 5
          if (h>270 and h<=295):
              QH = 6
          if (h>295 and h<=315):
              QH = 7
  
          QS = 0
          if (s>=0 and s<=0.2):
              QS = 0
          if (s>0.2 and s<=0.7):
              QS =1
          if (s>0.7 and s<=1):
              QS =2
  
          QV = 0
          if (v>=0 and v<=0.2):
              QV = 0
          if (v>0.2 and v<=0.7):
              QV =1
          if (v>0.7 and v<=1):
              QV =2
  
          L = 9 * QH +3 * QS +QV
          assert  L>=0 and L<=71
          return  L
  
  
      def _getImageColorVector(self, image):
          originImage = Image.open(image)
          ndarr = np.array(originImage.convert("RGB"))
          rowcnt = ndarr.shape[0]
          colcnt = ndarr.shape[1]
          #colors = ndarr.shapr[2]
  
          LVector =  [0] * 12
  
          for oneRow in range(rowcnt):
              for  oneCol in range(colcnt):
                  r, g, b = ndarr[oneRow][oneCol]
  
                  h, s, v = colorsys.rgb_to_hsv(r/255, g/255, b/255)
                  h = h * 360
                  l = self._hsv2L(h, s, v)
                  LVector[l/6] +=1
  
          lsum  =sum(LVector)
          result = [v * 1.0/lsum for v in LVector]
          print image, result
          return result
  
      def _getdata(self):
          date_file = open(self._imageVectorFile, 'r')
          imagesdir = []
          imagesdata = []
          for line in date_file:
              p = line.strip('\n').split(',')
              imagesdir.append(p[0])
              imagesdata.append(map(eval, p[1:]))
              self.coments = imagedir[:]
          return imagesdata
  
      def _getimagedir(self):
          date_file = open(self._imageVectorFile, 'r')
          imagesdir = []
          for line in date_file:
              p = line.strip('\n').split(',')
              imagesdir.append(p[0])
          return imagesdir
  
  
  
  
      def cluster(self):
          images = self._loadImages()
          file = open(self._imageVectorFile, 'w')
          for oneimage in images:
              if not os.path.isdir(oneimage):
                  lvector = self._getImageColorVector(oneimage)
                  file.write(oneimage+',')
                  file.write(','.join(map(str, lvector)))
                  file.write('\n')
          file.close()
          km = kmeans
          points = self._getdata()
          print points
          imageMembers = km.k_means(points, self._k)
          print imageMembers
          for idx, m in enumerate(imageMembers):
              srcdir = self._getimagedir()
              src = srcdir[idx]
              print 'clusterid:', m
              print 'src',src
              dest = os.path.join(os.path.join(self._imageDir), 'cluster-{%d}' % m)
              print dest
              if not os.path.exists(dest):
                  os.makedirs(dest)
              shutil.copy(src, dest)
  
  if __name__ == '__main__':
      basedir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
      imagedir =os.path.join(basedir, 'images')
      imageluster =ImagesCluster(imagedir, 3)
      imageluster.cluster()

  
  

  2