Matteo Sirri commited on
Commit
3e01e59
·
1 Parent(s): 23918e1

fix: fix typo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def load_model(baseline: bool = False):
22
  in_features = model.roi_heads.box_predictor.cls_score.in_features
23
  model.roi_heads.box_predictor = FastRCNNPredictor(in_features, 2)
24
  checkpoint = torch.load(
25
- osp.join(os.getcwd(),"model_split_3_FT_MOT17.pth"), map_location="cpu")
26
  model.load_state_dict(checkpoint["model"])
27
  model.to(device)
28
  model.eval()
 
22
  in_features = model.roi_heads.box_predictor.cls_score.in_features
23
  model.roi_heads.box_predictor = FastRCNNPredictor(in_features, 2)
24
  checkpoint = torch.load(
25
+ osp.join(os.getcwd(), "model_split3_FT_MOT17.pth"), map_location="cpu")
26
  model.load_state_dict(checkpoint["model"])
27
  model.to(device)
28
  model.eval()