Update app.py

app.py

@@ -579,6 +579,7 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
       bg = cv2.cvtColor(d["background"], cv2.COLOR_RGBA2RGB)
       bg = cv2.cvtColor(bg, cv2.COLOR_RGB2HSV)
       bg[:,:,0] = clrs[:,:,0]
+      bg = cv2.cvtColor(bg, cv2.COLOR_HSV2RGB)
 
       diff = np.abs(cv2.cvtColor(d["background"], cv2.COLOR_RGBA2GRAY).astype(np.int16)-cv2.cvtColor(gradient, cv2.COLOR_RGBA2GRAY).astype(np.int16)).astype(np.uint8)
       mask = cv2.inRange(diff, 0, t)
@@ -597,7 +598,7 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
       #    bg[(bg >= i*255/l) & (bg < (i+1)*255/l)] = i*255/(l-1)
       #bg = cv2.convertScaleAbs(palette[bg]).astype(np.uint8) # Converting image back to uint
 
-      res = np.float32(bg.reshape((-1,1)))
+      res = np.float32(bg.reshape((-1,3)))
       criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 4, 1.0)
       ret,label,center=cv2.kmeans(res,l,None,criteria,4,cv2.KMEANS_PP_CENTERS)
       center = np.uint8(center)
@@ -610,7 +611,6 @@ def draw_mask(l, t, b, v, d, evt: gr.EventData):
       params["fnum"] = frame_selected
       params["l"] = l
 
-      bg = cv2.cvtColor(bg, cv2.COLOR_HSV2RGB)
       d["layers"][0] = cv2.cvtColor(bg, cv2.COLOR_RGB2RGBA)
       edge = bg.copy()
     else: