timmhaucke commited on
Commit
11e6dd5
·
verified ·
1 Parent(s): 9349e97

Add weight files in correct directory

Browse files
Files changed (3) hide show
  1. app.py +1 -1
  2. weights/ear_YOLOv5_n.pt +3 -0
  3. weights/weights.pt +3 -0
app.py CHANGED
@@ -19,7 +19,7 @@ ear_detector.to(device)
19
  model = timm.create_model("hf-hub:BVRA/MegaDescriptor-T-224", pretrained=True, num_classes=0)
20
 
21
  # load state dict containing miscellaneous state or just the model weights
22
- state_dict = torch.load("weights.pt", map_location=device)
23
  if "optimizer" in state_dict:
24
  model.load_state_dict(state_dict["model"])
25
  else:
 
19
  model = timm.create_model("hf-hub:BVRA/MegaDescriptor-T-224", pretrained=True, num_classes=0)
20
 
21
  # load state dict containing miscellaneous state or just the model weights
22
+ state_dict = torch.load("weights/weights.pt", map_location=device)
23
  if "optimizer" in state_dict:
24
  model.load_state_dict(state_dict["model"])
25
  else:
weights/ear_YOLOv5_n.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e7aabe974a13c15ea2492a7f1583ba13255163305d0fc9c0aff8b8ee33da6651
3
+ size 3923125
weights/weights.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da82dcf2c50dec71cafefd803cc8d078cdcf9226cbf571b7adde0dd4b14c6e7a
3
+ size 224710970