Skip to content

Commit 2e755cc

Browse files
committed
Fix livecell patch shape in finetuning
1 parent 2291cbd commit 2e755cc

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,4 @@ __pycache__/
44
*.tif
55
examples/data/*
66
*.out
7+
*.png

finetuning/livecell_finetuning.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def finetune_livecell(args):
3838
# training settings:
3939
model_type = args.model_type
4040
checkpoint_path = None # override this to start training from a custom checkpoint
41-
patch_shape = (520, 740) # the patch shape for training
41+
patch_shape = (520, 704) # the patch shape for training
4242
n_objects_per_batch = 25 # this is the number of objects per batch that will be sampled
4343

4444
# get the trainable segment anything model

0 commit comments

Comments
 (0)