we were missing __init__.py in the models module

weecology · Dec 4, 2023 · dad1bfe · dad1bfe
1 parent 450e051
commit dad1bfe
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/deepforest/models/# Model modules b/deepforest/models/# Model modules
diff --git a/deepforest/models/__init__.py b/deepforest/models/__init__.py
diff --git a/deepforest/visualize.py b/deepforest/visualize.py
@@ -80,8 +80,10 @@ def plot_prediction_dataframe(df,
     Args:
         df: a pandas dataframe with image_path, xmin, xmax, ymin, ymax and label columns. The image_path column should be the relative path from root_dir, not the full path.
         root_dir: relative dir to look for image names from df.image_path
-        ground_truth: an optional pandas dataframe in same format as df holding ground_truth boxes
         savedir: save the plot to an optional directory path.
+        color: color of the bounding box as a tuple of BGR color, e.g. orange annotations is (0, 165, 255)
+        thickness: thickness of the rectangle border line in px
+        ground_truth: an optional pandas dataframe in same format as df holding ground_truth boxes
     Returns:
         written_figures: list of filenames written
         """