shvardhan commited on
Commit
58d1469
·
1 Parent(s): 6fbb3c0

update config file

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -23,7 +23,7 @@ class Model:
23
 
24
 
25
  def _load_model(self, name: str) -> nn.Module:
26
- return init_detector(model = 'faster-rcnn_r50_fpn_1x_coco',weights = 'models/orgaquant_pretrained.pth' , device=self.device)
27
 
28
  def set_model(self, name: str) -> None:
29
  if name == self.model_name:
 
23
 
24
 
25
  def _load_model(self, name: str) -> nn.Module:
26
+ return init_detector(config = 'https://github.com/open-mmlab/mmdetection/tree/master/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py',checkpoint = 'models/orgaquant_pretrained.pth' , device=self.device)
27
 
28
  def set_model(self, name: str) -> None:
29
  if name == self.model_name: