resolved numpy bug

.gitignore

@@ -1,4 +1,5 @@
 venv
 __pycache__
 uploaded_samples
-demo_data/results
+demo_data/results
+.idea

app.py

@@ -166,12 +166,13 @@ if st.paths is not None:
     target = np.squeeze(target)
     target = [classes_color_map[p] for p in target]
 
+
     fig, ax = plt.subplots()
     ax.imshow(target)
     
 
     markdown_legend = ''
-    for c, l in zip(classes_color_map, labels_map):
+    for c, l in zip(color_labels, labels_map):
         # print(colors.to_hex(c))
         markdown_legend += f'<div style="color:gray;background-color: {colors.to_hex(c)};">{l}</div><br>'
 

utils.py

@@ -658,7 +658,7 @@ class FPN(nn.Module):
 
 
 def ids_to_labels(dataloader, pred_labels):
-    new = np.zeros(pred_labels.shape, np.int)
+    new = np.zeros(pred_labels.shape, int)
     for cl, i in zip(dataloader.dataset.classids, range(len(dataloader.dataset.classids))):
         if type(cl) is list:
             new[pred_labels == i] = cl[0]
@@ -727,37 +727,25 @@ os.environ['CUDA_VISIBLE_DEVICES'] = opt.gpu_id
 if opt.no_cuda:
     os.environ['CUDA_VISIBLE_DEVICES'] = ""
 
-classes_color_map = np.array([[0., 0., 0.],                          # Black
-                              [0.79215686, 1.00000000, 0.43921569],  # DarkGreen
-                              [0.78039216, 0.08235294, 0.52156863],  # MediumVioletRed
-                              [0.00000000, 0.39215686, 0.00000000],  # DarkOliveGreen1
-                              [0, 0, 1],                             # Blue
-                              [1.00000000, 0.84313725, 0.00000000],  # gold
-                              [0.62745098, 0.12549020, 0.94117647],  # purple
-                              [0.0, 100/255, 0.0],                   # DarkGreen
-                              [0.98039216, 0.66666667, 0.62745098],  #
-                              [1, 1, 0],                             # yellow
-                              [0.2745098, 0.2745098, 0.2745098],     #
-                              [0, 250/255, 154/255],                 # MediumSpringGreen
-                              [64/255, 224/255, 208/255],            # turquoise
-                              [0.70588235, 0.64705882, 0.70588235],  #
-                              [0.58823529, 0.39215686, 0.39215686],  #
-                              [139/255, 35/255, 35/255],             # brown4
-                              [0.6, 0.6, 0.6],
-                              [0.98039216, 0.66666667, 0.11764706],
-                              [0.8627451, 0.8627451, 0.],
-                              [0.41960784, 0.55686275, 0.1372549],
-                              [0.2745098, 0.50980392, 0.70588235],
-                              [0.8627451, 0.07843137, 0.23529412],
-                              [1., 0., 0.],
-                              [0., 0., 0.55686275],
-                              [0., 0., 0.2745098],
-                              [0., 0.23529412, 0.39215686],
-                              [0., 0., 0.35294118],
-                              [0., 0., 0.43137255],
-                              [0., 0.31372549, 0.39215686],
-                              [0., 0., 0.90196078],
-                              [0.46666667, 0.04313725, 0.1254902],
-                              [0., 0., 0.55686275]])
-colors = classes_color_map[:8] # num classes
-labels_map = ['Unknown', 'Other cereals', 'Woods and other tree crops', 'Forage', 'Corn', 'Rice', 'Unknow cropland', 'No arable land']
+classes_color_map = np.array([[255/255.0, 255/255.0, 255/255.0],
+                              [150/255.0, 150/255.0, 150/255.0],  # Gray   - 1 - Unknow cropland
+                              [255/255.0, 0, 0],      # Red    - 2 - Other cereals
+                              [161/255.0, 0, 0],
+                              [0., 255/255.0, 0],     # Green  - 4 - Woods and other tree crops
+                              [255/255.0, 240/255.0, 0],
+                              [130/255.0, 120/255.0, 240/255.0],
+                              [255/255.0, 136/255.0, 0],    # Orange - 7 - Forage
+                              [250, 190, 190],
+                              [255/255.0, 255/255.0, 0],    # Yellow - 9 - Corn
+                              [0, 250, 154],
+                              [64, 224, 208],
+                              [0/255.0, 255/255.0, 255/255.0],    # Turchese - 12 - Rice
+                              [0.58823529, 0.39215686, 0.39215686],
+                              [139/255, 35/255, 35/255],
+                              [139/255, 35/255, 35/255],
+                              [139/255, 35/255, 35/255],
+                              [0, 0, 0],        # Black - 17 -  No Arable land
+                              ])
+color_labels = [classes_color_map[1], classes_color_map[2], classes_color_map[4], classes_color_map[7],
+                classes_color_map[9], classes_color_map[12], classes_color_map[17]]
+labels_map = ['Unknow cropland', 'Other cereals', 'Woods and other tree crops', 'Forage', 'Corn', 'Rice', 'No arable land']