yucornetto commited on
Commit
6bd742d
1 Parent(s): f57c31f

init for demo

Browse files

Change-Id: Ie96c21f2c564384fccc97e4547d5d409f69f7502

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -47,9 +47,9 @@ add_fcclip_config(cfg)
47
  cfg.merge_from_file("configs/coco/panoptic-segmentation/fcclip/fcclip_convnext_large_eval_ade20k.yaml")
48
  os.system("gdown 1-91PIns86vyNaL3CzMmDD39zKGnPMtvj")
49
  cfg.MODEL.WEIGHTS = './fcclip_cocopan.pth'
50
- cfg.MODEL.KMAX_DEEPLAB.TEST.SEMANTIC_ON = False
51
- cfg.MODEL.KMAX_DEEPLAB.TEST.INSTANCE_ON = False
52
- cfg.MODEL.KMAX_DEEPLAB.TEST.PANOPTIC_ON = True
53
  predictor = DefaultPredictor(cfg)
54
 
55
  # def inference(img):
 
47
  cfg.merge_from_file("configs/coco/panoptic-segmentation/fcclip/fcclip_convnext_large_eval_ade20k.yaml")
48
  os.system("gdown 1-91PIns86vyNaL3CzMmDD39zKGnPMtvj")
49
  cfg.MODEL.WEIGHTS = './fcclip_cocopan.pth'
50
+ cfg.MODEL.MASK_FORMER.TEST.SEMANTIC_ON = False
51
+ cfg.MODEL.MASK_FORMER.TEST.INSTANCE_ON = False
52
+ cfg.MODEL.MASK_FORMER.TEST.PANOPTIC_ON = True
53
  predictor = DefaultPredictor(cfg)
54
 
55
  # def inference(img):