| { | |
| "architectures": [ | |
| "CondUNetForSemanticSegmentation" | |
| ], | |
| "attn_start": 0, | |
| "auto_map": { | |
| "AutoConfig": "configuration_cond_unet.CondUNetConfig", | |
| "AutoImageProcessor": "image_processing_cond_unet.CondUNetImageProcessor", | |
| "AutoModelForImageSegmentation": "modeling_cond_unet.CondUNetForSemanticSegmentation" | |
| }, | |
| "custom_pipelines": { | |
| "image-segmentation": { | |
| "impl": "pipeline.CondUNetImageSegmentationPipeline", | |
| "pt": [ | |
| "AutoModelForImageSegmentation" | |
| ], | |
| "type": "image" | |
| } | |
| }, | |
| "depth": 5, | |
| "dtype": "float32", | |
| "dwt_bands": [ | |
| "LL", | |
| "LH", | |
| "HL", | |
| "HH" | |
| ], | |
| "emb_dim": 768, | |
| "id2label": { | |
| "0": "foreground" | |
| }, | |
| "image_size": 512, | |
| "in_channels": 3, | |
| "keep_aspect_ratio": true, | |
| "label2id": { | |
| "foreground": 0 | |
| }, | |
| "model_type": "cond_unet", | |
| "n_heads": 8, | |
| "n_organs": 10, | |
| "patch_size": 8, | |
| "self_normalize": true, | |
| "shape_res": 32, | |
| "size": 16, | |
| "transformers_version": "5.16.1", | |
| "unknown_organ_id": -1, | |
| "use_attn": true, | |
| "use_dwt": false, | |
| "use_shape": false, | |
| "wavelet": "haar" | |
| } | |