Skip to content
Snippets Groups Projects
Commit 8b6f7b03 authored by alan's avatar alan
Browse files

refactor(tests): get_traps->get_tiles

parent 0a09e682
No related branches found
No related tags found
No related merge requests found
...@@ -27,7 +27,7 @@ plt.show() ...@@ -27,7 +27,7 @@ plt.show()
tiler = Tiler(expt, template=trap_template) tiler = Tiler(expt, template=trap_template)
# Load images (takes about 5 mins) # Load images (takes about 5 mins)
trap_tps = tiler.get_traps_timepoint(0, tile_size=117, z=[2]) trap_tps = tiler.get_tiles_timepoint(0, tile_size=117, z=[2])
# Plot found traps # Plot found traps
nrows, ncols = (5, 5) nrows, ncols = (5, 5)
......
...@@ -42,7 +42,7 @@ def test_client(): ...@@ -42,7 +42,7 @@ def test_client():
num_timepoints = 5 num_timepoints = 5
traps_tps = [ traps_tps = [
seg_expt.get_traps_timepoint( seg_expt.get_tiles_timepoint(
tp, tile_size=81, channels=[0], z=[0, 1, 2, 3, 4] tp, tile_size=81, channels=[0], z=[0, 1, 2, 3, 4]
).squeeze() ).squeeze()
for tp in range(num_timepoints) for tp in range(num_timepoints)
......
...@@ -43,7 +43,7 @@ def trap_timelapse(tiler, trap_idx, channel, z): ...@@ -43,7 +43,7 @@ def trap_timelapse(tiler, trap_idx, channel, z):
def timepoint_traps(tiler, tp_idx, channel, z, tile_size): def timepoint_traps(tiler, tp_idx, channel, z, tile_size):
channel_id = tiler.get_channel_index(channel) channel_id = tiler.get_channel_index(channel)
traps = tiler.get_traps_timepoint( traps = tiler.get_tiles_timepoint(
tp_idx, tile_size=tile_size, channels=[channel_id], z=list(range(z)) tp_idx, tile_size=tile_size, channels=[channel_id], z=list(range(z))
) )
return traps return traps
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment