Update callbacks.py

DIAGNijmegen · Jun 23, 2024 · 826aad1 · 826aad1
1 parent 6e12013
commit 826aad1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wholeslidedata/annotation/callbacks.py b/wholeslidedata/annotation/callbacks.py
@@ -55,12 +55,12 @@ def __call__(self, annotations: List[Annotation]):
             for x in range(x1, x2, self._tile_size-self._overlap):
                 for y in range(y1, y2, self._tile_size-self._overlap):
                     box_poly = geometry.box(x, y, x+self._tile_size, y+self._tile_size)
-                    if not self._full_coverage or box_poly.within(annotation):
+                    if not self._full_coverage or box_poly.within(annotation.geometry):
                         new_annotations.append(Annotation.create(
                             index=index,
                             coordinates=box_poly.exterior.coords,
                             label=annotation.label.todict(),
                                 ))
                         index += 1
 
-        return new_annotations
+        return new_annotations