Skip to content

Commit

Permalink
🐛 Add IOConfig for NuClick in pretrained_model.yaml (#709)
Browse files Browse the repository at this point in the history
- Add `IOConfig` for NuClick in `pretrained_model.yaml`
  • Loading branch information
mostafajahanifar authored Sep 1, 2023
1 parent e7e5de5 commit eb49f66
Showing 1 changed file with 22 additions and 2 deletions.
24 changes: 22 additions & 2 deletions tiatoolbox/data/pretrained_model.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -868,6 +868,16 @@ nuclick_original-pannuke:
kwargs:
num_input_channels: 5
num_output_channels: 1
ioconfig:
class: semantic_segmentor.IOSegmentorConfig
kwargs:
input_resolutions:
- {'units': 'baseline', 'resolution': 0.25}
output_resolutions:
- {'units': 'baseline', 'resolution': 0.25}
patch_input_shape: [128, 128]
patch_output_shape: [128, 128]
save_resolution: {'units': 'baseline', 'resolution': 1.0}

nuclick_light-pannuke:
url: https://tiatoolbox.dcs.warwick.ac.uk/models/seg/nuclick_unet-pannuke.pth
Expand All @@ -878,5 +888,15 @@ nuclick_light-pannuke:
num_output_channels: 1
encoder: "unet"
encoder_levels: [32, 64, 128, 256]
decoder_block: (3,)
skip_type: 'concat'
decoder_block: [3,3]
skip_type: "add"
ioconfig:
class: semantic_segmentor.IOSegmentorConfig
kwargs:
input_resolutions:
- {'units': 'baseline', 'resolution': 0.25}
output_resolutions:
- {'units': 'baseline', 'resolution': 0.25}
patch_input_shape: [128, 128]
patch_output_shape: [128, 128]
save_resolution: {'units': 'baseline', 'resolution': 1.0}

0 comments on commit eb49f66

Please sign in to comment.