Skip to content

Commit

Permalink
fix(stitch prediction): split sample tiles correctly
Browse files Browse the repository at this point in the history
  • Loading branch information
melisande-c committed Jun 24, 2024
1 parent ca55b3c commit 31bd146
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/careamics/prediction_utils/stitch_prediction.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,9 @@ def stitch_prediction(
last_tiles = [tile_info.last_tile for tile_info in tile_infos]
last_tile_position = np.where(last_tiles)[0]
image_slices = [
slice(None if i == 0 else last_tile_position[i - 1], last_tile_position[i] + 1)
slice(
None if i == 0 else last_tile_position[i - 1] + 1, last_tile_position[i] + 1
)
for i in range(len(last_tile_position))
]
image_predictions = []
Expand Down

1 comment on commit 31bd146

@melisande-c
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Oops pushed to main by mistake! Will make a real PR

Please sign in to comment.