Hugging Face's logo

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

App Files Community
hlydecker commited on
Commit
db42787
•
1 Parent(s): be24e47

Update backend.py

Browse files
Files changed (1) hide show
  1. backend.py +2 -2
backend.py CHANGED
@@ -30,9 +30,9 @@ 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_cfg.yml")
34
  tree_cfg.MODEL.DEVICE='cpu'
35
- tree_cfg.MODEL.WEIGHTS = "tree_model_weights/treev1_best.pth"
36
  tree_cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
37
  tree_predictor = DefaultPredictor(tree_cfg)
38
 
 
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
37
  tree_predictor = DefaultPredictor(tree_cfg)
38