tf: update docs

mneher11 · Oct 15, 2020 · ea0d54b · ea0d54b
1 parent 0e73c11
commit ea0d54b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/py_src/yolov4/tf/__init__.py b/py_src/yolov4/tf/__init__.py
@@ -283,6 +283,13 @@ def save_dataset_for_mAP(
         """
         gt: name left top right bottom
         dr: name confidence left top right bottom
+
+        @param `mAP_path`
+        @parma `data_set`
+        @param `num_sample`: Number of images for mAP. If `None`, all images in
+                `data_set` are used.
+        @parma `images_optional`: If `True`, images are copied to the
+                `mAP_path`.
         """
         input_path = path.join(mAP_path, "input")