apply new style of display to function 'display'

Author: zylo117

utils/utils.py

@@ -136,13 +136,11 @@ def display(preds, imgs, obj_list, imshow=True, imwrite=False):
 
         for j in range(len(preds[i]['rois'])):
             (x1, y1, x2, y2) = preds[i]['rois'][j].astype(np.int)
-            cv2.rectangle(imgs[i], (x1, y1), (x2, y2), (255, 255, 0), 2)
             obj = obj_list[preds[i]['class_ids'][j]]
             score = float(preds[i]['scores'][j])
 
-            cv2.putText(imgs[i], '{}, {:.3f}'.format(obj, score),
-                        (x1, y1 + 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5,
-                        (255, 255, 0), 1)
+            plot_one_box(imgs[i], [x1, y1, x2, y2], label=obj, score=score,
+                         color=color_list[get_index_label(obj, obj_list)])
         if imshow:
             cv2.imshow('img', imgs[i])
             cv2.waitKey(0)
@@ -245,6 +243,7 @@ def variance_scaling_(tensor, gain=1.):
 
     return _no_grad_normal_(tensor, 0., std)
 
+
 STANDARD_COLORS = [
     'LawnGreen', 'Chartreuse', 'Aqua','Beige', 'Azure','BlanchedAlmond','Bisque',
     'Aquamarine', 'BlueViolet', 'BurlyWood', 'CadetBlue', 'AntiqueWhite',
@@ -271,21 +270,25 @@ def variance_scaling_(tensor, gain=1.):
     'WhiteSmoke', 'Yellow', 'YellowGreen'
 ]
 
+
 def from_colorname_to_bgr(color):
     rgb_color=webcolors.name_to_rgb(color)
     result=(rgb_color.blue,rgb_color.green,rgb_color.red)
     return result
 
+
 def standard_to_bgr(list_color_name):
     standard= []
     for i in range(len(list_color_name)-36): #-36 used to match the len(obj_list)
         standard.append(from_colorname_to_bgr(list_color_name[i]))
     return standard
 
+
 def get_index_label(label, obj_list):
     index = int(obj_list.index(label))
     return index
 
+
 def plot_one_box(img, coord, label=None, score=None, color=None, line_thickness=None):
     tl = line_thickness or int(round(0.001 * max(img.shape[0:2])))  # line thickness
     color = color
@@ -298,3 +301,6 @@ def plot_one_box(img, coord, label=None, score=None, color=None, line_thickness=
         c2 = c1[0] + t_size[0]+s_size[0]+15, c1[1] - t_size[1] -3
         cv2.rectangle(img, c1, c2 , color, -1)  # filled
         cv2.putText(img, '{}: {:.0%}'.format(label, score), (c1[0],c1[1] - 2), 0, float(tl) / 3, [0, 0, 0], thickness=tf, lineType=cv2.FONT_HERSHEY_SIMPLEX)
+
+        
+color_list = standard_to_bgr(STANDARD_COLORS)