Segmentation palette modifications (#3053)

* Segmentation palette modifications

* Embed color profile for palette

PythonClient/computer_vision/seg_pallete.py → PythonClient/computer_vision/seg_palette.py

@@ -1,5 +1,6 @@
 import numpy
 import random
+
 # requires Python 3.5.3 :: Anaconda 4.4.0
 # pip install opencv-python
 import cv2
@@ -13,24 +14,35 @@ def generate_color_palette(numPixelsWide, outputFile):
     possibilities = [list(range(256)), list(range(256)), list(range(256))]
 
     colors = [[0] * 3 for i in range(256)]
-
+
+    choice = 0
+    j = 0
+    for i in range(3):
+        palette[0, j * numPixelsWide : (j + 1) * numPixelsWide, i] = choice
+        colors[j][i] = choice
+
     for i in range(3):
-        for j in range(256):
+        for j in range(1, 255):
             choice = random.sample(possibilities[i], 1)[0]
             possibilities[i].remove(choice)
-            palette[0, j * numPixelsWide:(j + 1) * numPixelsWide, i] = choice
+            palette[0, j * numPixelsWide : (j + 1) * numPixelsWide, i] = choice
             colors[j][i] = choice
 
-    cv2.imwrite(outputFile, palette, [cv2.IMWRITE_PNG_COMPRESSION,0])
+    choice = 255
+    j = 255
+    for i in range(3):
+        palette[0, j * numPixelsWide : (j + 1) * numPixelsWide, i] = choice
+        colors[j][i] = choice
+
+    cv2.imwrite(outputFile, palette, [cv2.IMWRITE_PNG_COMPRESSION, 0])
 
     rgb_file = open("rgbs.txt", "w")
     for j in range(256):
         rgb_file.write("%d\t%s\n" % (j, str(list(reversed(colors[j])))))
     rgb_file.close()
 
 
-if __name__ == '__main__':
-
+if __name__ == "__main__":
     numPixelsWide = 4
-    outputFile = 'seg_color_pallet.png'
-    generate_color_palette(numPixelsWide, outputFile)
+    outputFile = "seg_color_palette.png"
+    generate_color_palette(numPixelsWide, outputFile)

docs/seg_rgbs.txt

@@ -1,4 +1,4 @@
-0	[55, 181, 57]
+0	[0, 0, 0]
 1	[153, 108, 6]
 2	[112, 105, 191]
 3	[89, 121, 72]
@@ -253,4 +253,4 @@
 252	[130, 56, 55]
 253	[147, 210, 11]
 254	[162, 203, 118]
-255	[43, 47, 206]
+255	[255, 255, 255]