Skip to content

Instantly share code, notes, and snippets.

@rekalantar

rekalantar/medsegmentanything_finetuning.py

Created May 30, 2023 10:18
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save rekalantar/430fae5a25163ea071afffe38696db2a to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save rekalantar/430fae5a25163ea071afffe38696db2a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
medsegmentanything_finetuning.py
# create train and validation dataloaders
train_dataset = SAMDataset(image_paths=data_paths['train_images'], mask_paths=data_paths['train_masks'], processor=processor)
train_dataloader = DataLoader(train_dataset, batch_size=1, shuffle=True)
val_dataset = SAMDataset(image_paths=data_paths['val_images'], mask_paths=data_paths['val_masks'], processor=processor)
val_dataloader = DataLoader(val_dataset, batch_size=1, shuffle=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment