Hugging Face's logo

Spaces:
SIH
/
aerial-segmentation-model-selection
Runtime error

App Files Community
hlydecker commited on
Commit
e793dfa
•
1 Parent(s): cf86d04

patch: prayer

Browse files
Files changed (1) hide show
  1. backend.py +1 -1
backend.py CHANGED
@@ -30,7 +30,7 @@ from detectron2.structures import Instances
30
 
31
  # Model for trees
32
  tree_cfg = get_cfg()
33
- tree_cfg.merge_from_file("tree_model_weights/tree_v2_cfg.yml")
34
  tree_cfg.MODEL.DEVICE='cpu'
35
  tree_cfg.MODEL.WEIGHTS = "tree_model_weights/treev2_best.pth"
36
  tree_cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
 
30
 
31
  # Model for trees
32
  tree_cfg = get_cfg()
33
+ tree_cfg.merge_from_file("tree_model_weights/treev2_cfg.yml")
34
  tree_cfg.MODEL.DEVICE='cpu'
35
  tree_cfg.MODEL.WEIGHTS = "tree_model_weights/treev2_best.pth"
36
  tree_cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2