Skip to content

Commit

Permalink
Remove debug statement
Browse files Browse the repository at this point in the history
  • Loading branch information
jonaraphael committed Oct 31, 2024
1 parent f66d8fc commit 7800705
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cerulean_cloud/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -865,7 +865,6 @@ def stitch(
scene_array_probs = np.zeros(
(num_classes, final_height, final_width), dtype=sample_tile.dtype
)
print("DEBUG: scene_array_probs.shape", scene_array_probs.shape)
# Place each tile into the final array
for tile_probs, bounds in zip(tile_probs_by_class, bounds_list):
x_offset = int(round((bounds[0] - min_x) / res_x))
Expand Down Expand Up @@ -897,6 +896,7 @@ def instantiate(self, scene_array_probs, transform):
# If there aren't any probability detections in the scene, also catches all-zeros scene_array_probs
if not np.any(scene_array_probs):
return geojson.FeatureCollection(features=[])

# Ensure scene_array_probs is a NumPy array
if isinstance(scene_array_probs, torch.Tensor):
scene_array_probs = scene_array_probs.detach().cpu().numpy()
Expand Down

0 comments on commit 7800705

Please sign in to comment.