Skip to content

Commit eb49f66

Browse files
🐛 Add IOConfig for NuClick in pretrained_model.yaml (#709)
- Add `IOConfig` for NuClick in `pretrained_model.yaml`
1 parent e7e5de5 commit eb49f66

File tree

1 file changed

+22
-2
lines changed

1 file changed

+22
-2
lines changed

tiatoolbox/data/pretrained_model.yaml

Lines changed: 22 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -868,6 +868,16 @@ nuclick_original-pannuke:
868868
kwargs:
869869
num_input_channels: 5
870870
num_output_channels: 1
871+
ioconfig:
872+
class: semantic_segmentor.IOSegmentorConfig
873+
kwargs:
874+
input_resolutions:
875+
- {'units': 'baseline', 'resolution': 0.25}
876+
output_resolutions:
877+
- {'units': 'baseline', 'resolution': 0.25}
878+
patch_input_shape: [128, 128]
879+
patch_output_shape: [128, 128]
880+
save_resolution: {'units': 'baseline', 'resolution': 1.0}
871881

872882
nuclick_light-pannuke:
873883
url: https://tiatoolbox.dcs.warwick.ac.uk/models/seg/nuclick_unet-pannuke.pth
@@ -878,5 +888,15 @@ nuclick_light-pannuke:
878888
num_output_channels: 1
879889
encoder: "unet"
880890
encoder_levels: [32, 64, 128, 256]
881-
decoder_block: (3,)
882-
skip_type: 'concat'
891+
decoder_block: [3,3]
892+
skip_type: "add"
893+
ioconfig:
894+
class: semantic_segmentor.IOSegmentorConfig
895+
kwargs:
896+
input_resolutions:
897+
- {'units': 'baseline', 'resolution': 0.25}
898+
output_resolutions:
899+
- {'units': 'baseline', 'resolution': 0.25}
900+
patch_input_shape: [128, 128]
901+
patch_output_shape: [128, 128]
902+
save_resolution: {'units': 'baseline', 'resolution': 1.0}

0 commit comments

Comments
 (0)