draw box with is_rescale aug, update docs for prepro

Author: zsdonghao

tensorlayer/prepro.py

@@ -1621,7 +1621,7 @@ def obj_box_left_right_flip(im, coords=[], is_rescale=False, is_center=False, is
         An image with dimension of [row, col, channel] (default).
     coords : list of list for coordinates [[x, y, w, h], [x, y, w, h], ...]
     is_rescale : boolean, default False
-        Set to True, if the coordinates are rescaled to [0, 1].
+        Set to True, if the input coordinates are rescaled to [0, 1].
     is_center : boolean, default False
         Set to True, if the x and y of coordinates are the centroid. (i.e. darknet format)
     is_random : boolean, default False
@@ -1700,7 +1700,7 @@ def obj_box_imresize(im, coords=[], size=[100, 100], interp='bicubic', mode=None
     coords : list of list for coordinates [[x, y, w, h], [x, y, w, h], ...]
     size, interp, mode : see ``tl.prepro.imresize`` for details.
     is_rescale : boolean, default False
-        Set to True, if the coordinates are rescaled to [0, 1], then return the original coordinates.
+        Set to True, if the input coordinates are rescaled to [0, 1], then return the original coordinates.
 
     Examples
     --------
@@ -1770,7 +1770,7 @@ def obj_box_crop(im, classes=[], coords=[], wrg=100, hrg=100,
     coords : list of list for coordinates [[x, y, w, h], [x, y, w, h], ...]
     wrg, hrg, is_random : see ``tl.prepro.crop`` for details.
     is_rescale : boolean, default False
-        Set to True, if the coordinates are rescaled to [0, 1].
+        Set to True, if the input coordinates are rescaled to [0, 1].
     is_center : boolean, default False
         Set to True, if the x and y of coordinates are the centroid. (i.e. darknet format)
     thresh_wh : float
@@ -1897,7 +1897,7 @@ def obj_box_shift(im, classes=[], coords=[], wrg=0.1, hrg=0.1,
     coords : list of list for coordinates [[x, y, w, h], [x, y, w, h], ...]
     wrg, hrg, row_index, col_index, channel_index, is_random, fill_mode, cval, order : see ``tl.prepro.shift``.
     is_rescale : boolean, default False
-        Set to True, if the coordinates are rescaled to [0, 1].
+        Set to True, if the input coordinates are rescaled to [0, 1].
     is_center : boolean, default False
         Set to True, if the x and y of coordinates are the centroid. (i.e. darknet format)
     thresh_wh : float
@@ -2007,7 +2007,7 @@ def obj_box_zoom(im, classes=[], coords=[], zoom_range=(0.9, 1.1),
     coords : list of list for coordinates [[x, y, w, h], [x, y, w, h], ...]
     zoom_range, row_index, col_index, channel_index, is_random, fill_mode, cval, order : see ``tl.prepro.zoom``.
     is_rescale : boolean, default False
-        Set to True, if the coordinates are rescaled to [0, 1].
+        Set to True, if the input coordinates are rescaled to [0, 1].
     is_center : boolean, default False
         Set to True, if the x and y of coordinates are the centroid. (i.e. darknet format)
     thresh_wh : float

tensorlayer/visualize.py

@@ -90,18 +90,18 @@ 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, save_name=None):
+                box_centroid_to_upleft_butright=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
     -----------
     image : RGB image in numpy.array, [height, width, channel].
-    classes : list of class ID (int).
-    coords : list of list for coordinates.
-        - [x, y, x2, y2] (up-left and botton-right)
-        - or [x_center, y_center, w, h] (set box_centroid_to_upleft_butright to True).
-    scores : list of score (int). (Optional)
-    classes_list : list of string, for converting ID to string.
+    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).
+    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.
         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.