AI-Cyber commited on
Commit
ca459a1
·
1 Parent(s): a89a43b

Update visual_utils.py

Browse files
Files changed (1) hide show
  1. visual_utils.py +6 -3
visual_utils.py CHANGED
@@ -58,10 +58,10 @@ def unify_Vai_cmap():
58
  colors = [
59
  [0, 0, 0],#0
60
  [0, 255, 0],
61
- [0, 0, 255],
62
  [0, 0, 255],
63
  [0, 0, 0],
64
- [159,129,183],
65
  [0, 255, 255], #6
66
  [255,195,128],
67
  [0, 0, 0],
@@ -70,7 +70,7 @@ def unify_Vai_cmap():
70
  [0, 0, 0],
71
  [0,0,0],#12
72
  [0, 0, 0],
73
- [255,0,0], #14
74
 
75
  ]
76
  for i in range(len(colors)):
@@ -161,6 +161,8 @@ def unify_cmap():
161
 
162
  return cmap.astype(np.uint8)
163
 
 
 
164
  def unify_YIJISAR_cmap():
165
  cmap = np.zeros((255, 3), dtype=np.uint8)
166
  colors = [
@@ -197,6 +199,7 @@ def unify_YIJISAR_cmap():
197
  return cmap.astype(np.uint8)
198
 
199
 
 
200
  def unify_single_cmap():
201
  cmap = np.zeros((255, 3), dtype=np.uint8)
202
  colors = [
 
58
  colors = [
59
  [0, 0, 0],#0
60
  [0, 255, 0],
61
+ [255, 255, 255],
62
  [0, 0, 255],
63
  [0, 0, 0],
64
+ [255,255,0],
65
  [0, 255, 255], #6
66
  [255,195,128],
67
  [0, 0, 0],
 
70
  [0, 0, 0],
71
  [0,0,0],#12
72
  [0, 0, 0],
73
+ [0,0,255], #14
74
 
75
  ]
76
  for i in range(len(colors)):
 
161
 
162
  return cmap.astype(np.uint8)
163
 
164
+
165
+
166
  def unify_YIJISAR_cmap():
167
  cmap = np.zeros((255, 3), dtype=np.uint8)
168
  colors = [
 
199
  return cmap.astype(np.uint8)
200
 
201
 
202
+
203
  def unify_single_cmap():
204
  cmap = np.zeros((255, 3), dtype=np.uint8)
205
  colors = [