Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
Spaces:
tgieruc
/
agnostic-segmentation
like
0
Runtime error
App
Files
Files
Community
main
agnostic-segmentation
1 contributor
History:
5 commits
tgieruc
loads weights to cpu if cuda not available
6d84c7f
over 1 year ago
.gitattributes
1.48 kB
initial commit
over 1 year ago
README.md
240 Bytes
initial commit
over 1 year ago
app.py
1.98 kB
uploaded space
over 1 year ago
demo_img.jpg
95.2 kB
uploaded space
over 1 year ago
model.py
7.64 kB
loads weights to cpu if cuda not available
over 1 year ago
model_segmentation.pth
pickle
Detected Pickle imports (23)
"torch.nn.modules.conv.Conv2d"
,
"__builtin__.set"
,
"torch._utils._rebuild_tensor_v2"
,
"segmentation_models_pytorch.decoders.unet.model.Unet"
,
"torch.nn.modules.batchnorm.BatchNorm2d"
,
"segmentation_models_pytorch.encoders.resnet.ResNetEncoder"
,
"segmentation_models_pytorch.base.heads.SegmentationHead"
,
"torch.nn.modules.linear.Identity"
,
"torchvision.models.resnet.BasicBlock"
,
"segmentation_models_pytorch.base.modules.Attention"
,
"torch._utils._rebuild_parameter"
,
"segmentation_models_pytorch.base.modules.Conv2dReLU"
,
"torch.nn.modules.container.ModuleList"
,
"segmentation_models_pytorch.decoders.unet.decoder.UnetDecoder"
,
"collections.OrderedDict"
,
"torch.LongStorage"
,
"torch.nn.modules.activation.ReLU"
,
"torch.nn.modules.pooling.MaxPool2d"
,
"segmentation_models_pytorch.base.modules.Activation"
,
"torch.nn.modules.container.Sequential"
,
"torch.nn.modules.activation.Sigmoid"
,
"torch.FloatStorage"
,
"segmentation_models_pytorch.decoders.unet.decoder.DecoderBlock"
How to fix it?
97.9 MB
LFS
uploaded space
over 1 year ago
requirements.txt
72 Bytes
added segmentation_models_pytorch to requirements.txt
over 1 year ago