semantic-10

Sleeping

adpro commited on Feb 10, 2023

Commit

af6cfb2

1 Parent(s): e000318

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,3 +1,4 @@
 import gluoncv
 import mxnet as mx
 from gluoncv.utils.viz import get_color_pallete
@@ -9,7 +10,7 @@ from gluoncv.data.transforms.presets.segmentation import test_transform
 # using cpu
 ctx = mx.cpu(0)
-model = gluoncv.model_zoo.get_model("deeplab_resnest269_ade", pretrained=True)
 def segmentation(image):

+import torch.hub
 import gluoncv
 import mxnet as mx
 from gluoncv.utils.viz import get_color_pallete
 # using cpu
 ctx = mx.cpu(0)
+model = torch.hub.load("kazuto1011/deeplab-pytorch", "deeplabv2_resnet101", pretrained='cocostuff164k', n_classes=182)
 def segmentation(image):