Update app.py

app.py

@@ -392,20 +392,28 @@ def optimize(v, d):
         scale = 1
         delta = 0
         ddepth = cv2.CV_16S
+        l = 48
     
         indices = np.arange(0,256)   # List of all colors 
-        divider = np.linspace(0,255,48+1)[1] # we get a divider
-        quantiz = np.int0(np.linspace(0,255,48)) # we get quantization colors
-        color_levels = np.clip(np.int0(indices/divider),0,48-1) # color levels 0,1,2..
+        divider = np.linspace(0,255,l+1)[1] # we get a divider
+        quantiz = np.int0(np.linspace(0,255,l)) # we get quantization colors
+        color_levels = np.clip(np.int0(indices/divider),0,l-1) # color levels 0,1,2..
         palette = quantiz[color_levels]
         for k, f in enumerate(frames):
             frame = cv2.imread(frames[k]).astype(np.uint8)
             frame = cv2.convertScaleAbs(palette[frame]).astype(np.uint8) # Converting image back to uint
             cv2.imwrite(frames[k], frame)
+            print(bincount(frame))
       return frames
     else:
       return depths
 
+def bincount(a):
+    a2D = a.reshape(-1,a.shape[-1])
+    col_range = (256, 256, 256) # generically : a2D.max(0)+1
+    a1D = np.ravel_multi_index(a2D.T, col_range)
+    return np.unravel_index(np.bincount(a1D).argmax(), col_range)
+
 def reset_mask():
     global frame_selected
     global masks