Сегментация изображения с использованием k-средних

Question

Я пытаюсь использовать алгоритм k-средних для задачи сегментации изображения. Проблема в том, что моя программа не сегментирует изображение. Не могли бы вы помочь мне найти ошибку в моем коде?

Фактически я установил число кластеров равным 32. Я использовал следующие структуры данных:

• 3 массива bleu,vert,rouge для хранения значений RGB для каждого пикселя

• 3 массива cluster_bleu,cluster_rouge,cluster_vert для хранения значения RGB для каждого кластера

• группа [i,0]=k отображает каждый пиксель i в кластер k

  import cv2
  
  import numpy
  
  import random
  def main():
      MAX_LARGEUR = 400
      MAX_HAUTEUR = 400
  
      K = 32 #Le fameux parametre K de l'algorithme
      imagecolor = cv2.imread('perr.jpg')
  
  
      if imagecolor.shape[0] > MAX_LARGEUR or imagecolor.shape[1] > MAX_HAUTEUR:
          factor1 = float(MAX_LARGEUR) / imagecolor.shape[0]
          factor2 = float(MAX_HAUTEUR) / imagecolor.shape[1]
          factor = min(factor1, factor2)
          imagecolor = cv2.resize(imagecolor, None, fx=factor, fy=factor, interpolation=cv2.INTER_AREA)
  
  
      nb_pixels = imagecolor.shape[0] * imagecolor.shape[1]
  
      bleu = imagecolor[:, :, 0].reshape(nb_pixels, 1)
      vert = imagecolor[:, :, 1].reshape(nb_pixels, 1)
      rouge = imagecolor[:, :, 2].reshape(nb_pixels, 1)
  
  
      cluster_bleu = numpy.zeros(K)
      cluster_vert = numpy.zeros(K)
      cluster_rouge = numpy.zeros(K)
  
      groupe = numpy.zeros((nb_pixels, 1)) 
  
      for i in range(0,K):
          groupe[i,0]=i
  
      for i in range(K,nb_pixels):
          groupe[i,0]=random.randint(0, K-1)
  
  
      condition =False
  
  
  
  
      def etape1(indices,i):
      s=indices.size
      rouge_s=0
      vert_s=0
      bleu_s=0
      #calcul de barycentre des points
      if s==0:
          cluster_rouge[i]=0  
          cluster_vert[i]=0
          cluster_bleu[i]=0
  
  
      if s >=1:
          for j in range(0,s):
              rouge_s=rouge_s+rouge[indices[j]]
              vert_s=vert_s+vert[indices[j]]
              bleu_s=bleu_s+bleu[indices[j]]
  
          #mise  jour des clusters 
  
          cluster_rouge[i]=rouge_s/s  
          cluster_vert[i]=vert_s/s
          cluster_bleu[i]=bleu_s/s        
  
  
      iteration=0
      oldGroupe = numpy.copy(groupe)
      while(condition==False) :
  
  
      for i in range(0,K):
  
          indices=numpy.where(groupe==i)[0]
          etape1(indices,i)
  
  
  
      for i in range(0,nb_pixels):
          minimum=10000;
          dist=0;
          index=-1;
          for j in range(0,K):
               dist=(cluster_rouge[j]-rouge[i])**2+(cluster_vert[j]-vert[i])**2+(cluster_bleu[j]-bleu[i])**2;
               if(dist<=minimum):
                  minimum=dist;
                  index=j;
  
  
  
          groupe[i,0]=index;
  
  
  
      condition=numpy.all(groupe==oldGroupe)
  
      oldGroupe = numpy.copy(groupe)  
  
  
      groupe=numpy.reshape(groupe, (imagecolor.shape[0], imagecolor.shape[1]))
  
  
  
  
      for i in range(0, imagecolor.shape[0]):
          for j in range(0, imagecolor.shape[1]):
              imagecolor[i,j,0] = (cluster_bleu[groupe[i,j]])
              imagecolor[i,j,1] = (cluster_vert[groupe[i,j]])
              imagecolor[i,j,2] = (cluster_rouge[groupe[i,j]])
  
      cv2.namedWindow("sortie")
      cv2.imshow("sortie", imagecolor)
      key = cv2.waitKey(0)
  if __name__ == "__main__":
      main()
  

Answer 1

Проблема в назначении oldGroupe=groupe; который не копирует массив, но создает ссылку с другим именем (oldGroupe), что указывает на те же данные, что и groupe, Таким образом, когда вы меняете groupe ты тоже меняешь oldGroupe, а также condition всегда верно.

Вы хотите создать копию данных в groupe с oldGroupe = numpy.copy(groupe),