Modelpublic

voitl/unet_plus_plus

sourceHugging Faceupdated 4y agoView on Hugging Face

0likes9downloads

config.json22 linesDownload Raw Back to root

1{2  "architectures": [3    "HFUnetPlusPlus"4  ],5  "auto_map": {6    "AutoConfig": "hf_config.UnetConfig",7    "AutoModelForImageSegmentation": "hf_model.HFUnetPlusPlus"8  },9  "decoder_channels": [10    1024,11    512,12    256,13    128,14    6415  ],16  "encoder_name": "resnet18",17  "input_channels": 1,18  "num_classes": 16,19  "torch_dtype": "float32",20  "transformers_version": "4.24.0"21}22