Hugging Face's logo

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

App Files Community
hlydecker commited on
Commit
3451b3f
•
1 Parent(s): 1b235be

Update backend.py

Browse files
Files changed (1) hide show
  1. backend.py +1 -1
backend.py CHANGED
@@ -32,7 +32,7 @@ from detectron2.structures import Instances
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
37
  tree_predictor = DefaultPredictor(tree_cfg)
38
 
 
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_final.pth"
36
  tree_cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
37
  tree_predictor = DefaultPredictor(tree_cfg)
38