style updates

weecology · Oct 2, 2023 · 0e73f87 · 0e73f87
1 parent a9ddc0a
commit 0e73f87
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/deepforest/main.py b/deepforest/main.py
@@ -117,7 +117,7 @@ def use_bird_release(self, check_release=True):
         self.config["score_thresh"] = 0.3
 
         # Set label dictionary to Bird
-        self.label_dict = {"Bird":0}
+        self.label_dict = {"Bird": 0}
         self.numeric_to_label_dict = {v: k for k, v in self.label_dict.items()}
 
     def create_model(self):

diff --git a/deepforest/preprocess.py b/deepforest/preprocess.py
@@ -78,13 +78,15 @@ def select_annotations(annotations, windows, index, allow_empty=False):
     offset = 40
     selected_annotations = annotations[(annotations.xmin > (window_xmin - offset)) &
                                        (annotations.xmin < (window_xmax)) &
-                                       (annotations.xmax > (window_xmin)) &
-                                       (annotations.ymin > (window_ymin - offset)) &
+                                       (annotations.xmax >
+                                        (window_xmin)) & (annotations.ymin >
+                                                          (window_ymin - offset)) &
                                        (annotations.xmax < (window_xmax + offset)) &
-                                       (annotations.ymin < (window_ymax)) &
-                                       (annotations.ymax > (window_ymin)) &
-                                       (annotations.ymax < (window_ymax + offset))].copy(
-                                           deep=True)
+                                       (annotations.ymin <
+                                        (window_ymax)) & (annotations.ymax >
+                                                          (window_ymin)) &
+                                       (annotations.ymax <
+                                        (window_ymax + offset))].copy(deep=True)
     # change the image name
     image_basename = os.path.splitext("{}".format(annotations.image_path.unique()[0]))[0]
     selected_annotations.image_path = "{}_{}.png".format(image_basename, index)