Segmentation image one hot encoding

이후 추가작성.

 

cls_label=np.zeros((24,label.shape[0],label.shape[1]))

         for i in range(label.shape[0]):

             for j in range(label.shape[1]):

                 for k in range(24):

                 class_n.append(np.logical_and(A[i][j],df.values[i]))

               

                     if np.all(label[i][j]==df.values[k]):

                         cls_label[k][i][j]=1

pixel하나하나 참조해가며 비교하니 매우 느리다.

하나의 for문에서 2차원 부분을 1차원으로 만들고 한번에 RGB에 대해서 비교해주니 훨씬 빠르다.

for i in range(len(df.index)):

            cls_label[i,:,:]=np.all(label.reshape((-1,3))==df.values[i],axis=1).reshape(cls_label.shape[1:3])