draw obj box box_centroid_to_upleft_butright --> is_center

Author: zsdonghao

tensorlayer/visualize.py

@@ -90,7 +90,7 @@ def imsave(images, size, path):
 # for object detection
 def draw_boxes_and_labels_to_image(image, classes=[], coords=[],
                 scores=[], classes_list=[],
-                box_centroid_to_upleft_butright=True, is_rescale=True, save_name=None):
+                is_center=True, is_rescale=True, save_name=None):
     """ Draw bboxes and class labels on image. Return or save the image with bboxes, example in the docs of ``tl.prepro``.
 
     Parameters
@@ -99,10 +99,10 @@ def draw_boxes_and_labels_to_image(image, classes=[], coords=[],
     classes : a list of class ID (int).
     coords : a list of list for coordinates.
         - Should be [x, y, x2, y2] (up-left and botton-right format)
-        - If [x_center, y_center, w, h] (set box_centroid_to_upleft_butright to True).
+        - If [x_center, y_center, w, h] (set is_center to True).
     scores : a list of score (float). (Optional)
     classes_list : list of string, for converting ID to string on image.
-    box_centroid_to_upleft_butright : boolean, defalt is True.
+    is_center : boolean, defalt is True.
         If coords is [x_center, y_center, w, h], set it to True for converting [x_center, y_center, w, h] to [x, y, x2, y2] (up-left and botton-right).
         If coords is [x1, x2, y1, y2], set it to False.
     is_rescale : boolean, defalt is True.
@@ -129,7 +129,7 @@ def draw_boxes_and_labels_to_image(image, classes=[], coords=[],
     thick = int((imh + imw) // 430)
 
     for i in range(len(coords)):
-        if box_centroid_to_upleft_butright:
+        if is_center:
             x, y, x2, y2 = prepro.obj_box_coord_centroid_to_upleft_butright(coords[i])
         else:
             x, y, x2, y2 = coords[i]