Specify unbiased shape while picking coordinates

funkelab · Sep 30, 2023 · b1ce717 · b1ce717
1 parent cc0964f
commit b1ce717
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/cellulus/criterions/oce_loss.py b/cellulus/criterions/oce_loss.py
@@ -34,7 +34,10 @@ def forward(self, prediction):
         if self.num_spatial_dims == 2:
             b, c, h, w = prediction.shape
 
-            num_anchors = int(self.density * h * w)
+            h_ = h - 2 * self.kappa  # unbiased height
+            w_ = w - 2 * self.kappa  # unbiased width
+
+            num_anchors = int(self.density * h_ * w_)
             anchor_coordinates_y = np.random.randint(
                 self.kappa, h - self.kappa, num_anchors
             )
@@ -46,7 +49,12 @@ def forward(self, prediction):
             )  # N x 2
         elif self.num_spatial_dims == 3:
             b, c, d, h, w = prediction.shape
-            num_anchors = int(self.density * d * h * w)
+
+            d_ = d - 2 * self.kappa  # unbiased depth
+            h_ = h - 2 * self.kappa  # unbiased height
+            w_ = w - 2 * self.kappa  # unbiased width
+
+            num_anchors = int(self.density * d_ * h_ * w_)
             anchor_coordinates_z = np.random.randint(
                 self.kappa, d - self.kappa, num_anchors
             )