Hugging Face
Models
Datasets
Spaces
Buckets
new
Docs
Enterprise
Pricing
Log In
Sign Up
smilelab
/
BrainSegFounder
like
1
Follow
SMILE Lab
8
Image Segmentation
medical
mri
brain-segmentation
self-supervised-learning
swin-transformer
monai
uk-biobank
brats
atlas
License:
uk-biobank-mta
Model card
Files
Files and versions
xet
Community
main
BrainSegFounder
1.4 GB
Ctrl+K
Ctrl+K
1 contributor
History:
11 commits
d3v-26
Remove test_models.py - for local use only
7475b01
verified
3 months ago
.gitattributes
Safe
1.52 kB
initial commit
3 months ago
README.md
8.71 kB
Upload README.md with huggingface_hub
3 months ago
SSL_Head.py
5.19 kB
Upload SSL_Head.py with huggingface_hub
3 months ago
model_weights_ATLAS-finetune.pt
pickle
Detected Pickle imports (34)
"torch.nn.modules.linear.Identity"
,
"__builtin__.set"
,
"torch.nn.modules.conv.ConvTranspose3d"
,
"monai.networks.blocks.dynunet_block.UnetOutBlock"
,
"torch.nn.modules.activation.LeakyReLU"
,
"torch.FloatStorage"
,
"torch.nn.modules.conv.Conv3d"
,
"monai.networks.nets.swin_unetr.PatchMerging"
,
"collections.OrderedDict"
,
"torch._utils._rebuild_parameter"
,
"monai.networks.blocks.convolutions.Convolution"
,
"torch.nn.modules.activation.Softmax"
,
"monai.networks.nets.swin_unetr.SwinUNETR"
,
"monai.networks.blocks.unetr_block.UnetrBasicBlock"
,
"monai.networks.blocks.patchembedding.PatchEmbed"
,
"torch.LongStorage"
,
"monai.networks.blocks.mlp.MLPBlock"
,
"torch.nn.modules.instancenorm.InstanceNorm3d"
,
"monai.networks.nets.swin_unetr.SwinTransformerBlock"
,
"monai.networks.blocks.unetr_block.UnetrUpBlock"
,
"torch.nn.modules.normalization.LayerNorm"
,
"torch.nn.modules.activation.GELU"
,
"monai.networks.nets.swin_unetr.BasicLayer"
,
"monai.networks.nets.swin_unetr.SwinTransformer"
,
"torch.nn.modules.linear.Linear"
,
"torch.nn.modules.container.ModuleList"
,
"torch.device"
,
"monai.networks.blocks.dynunet_block.UnetResBlock"
,
"__mp_main__.ATLASPredictor"
,
"torch.nn.parallel.distributed.DistributedDataParallel"
,
"torch.nn.modules.dropout.Dropout"
,
"monai.networks.nets.swin_unetr.WindowAttention"
,
"torch.distributed.algorithms.join._JoinConfig"
,
"torch._utils._rebuild_tensor_v2"
How to fix it?
409 MB
xet
Upload model_weights_ATLAS-finetune.pt with huggingface_hub
3 months ago
model_weights_ATLAS-pretrain.pt
pickle
Detected Pickle imports (4)
"torch.FloatStorage"
,
"collections.OrderedDict"
,
"torch.LongStorage"
,
"torch._utils._rebuild_tensor_v2"
What is a pickle import?
238 MB
xet
Upload model_weights_ATLAS-pretrain.pt with huggingface_hub
3 months ago
model_weights_BRATS-finetune.pt
pickle
Detected Pickle imports (7)
"collections.OrderedDict"
,
"_codecs.encode"
,
"torch.LongStorage"
,
"numpy.dtype"
,
"torch.FloatStorage"
,
"numpy.core.multiarray.scalar"
,
"torch._utils._rebuild_tensor_v2"
How to fix it?
256 MB
xet
Upload model_weights_BRATS-finetune.pt with huggingface_hub
3 months ago
model_weights_BRATS-pretrain.pt
pickle
Detected Pickle imports (4)
"torch.FloatStorage"
,
"collections.OrderedDict"
,
"torch.LongStorage"
,
"torch._utils._rebuild_tensor_v2"
What is a pickle import?
263 MB
xet
Upload model_weights_BRATS-pretrain.pt with huggingface_hub
3 months ago
model_weights_UKB-pretrain.pt
pickle
Detected Pickle imports (4)
"torch.FloatStorage"
,
"collections.OrderedDict"
,
"torch.LongStorage"
,
"torch._utils._rebuild_tensor_v2"
What is a pickle import?
237 MB
xet
Upload model_weights_UKB-pretrain.pt with huggingface_hub
3 months ago