sanket kheni commited on
Commit
1dc64a7
1 Parent(s): 26c05fc
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -22,12 +22,12 @@ retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/Reti
22
 
23
  from retina_model import *
24
 
25
- RetinaFace = load_model("retina_model/RetinaFace-Res50.h5",
26
- custom_objects={"FPN": FPN,
27
- "SSH": SSH,
28
- "BboxHead": BboxHead,
29
- "LandmarkHead": LandmarkHead,
30
- "ClassHead": ClassHead}
31
  )
32
 
33
  arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFace")
 
22
 
23
  from retina_model import *
24
 
25
+ RetinaFace = load_model("retina_model/RetinaFace-Res50.h5"
26
+ # custom_objects={"FPN": FPN,
27
+ # "SSH": SSH,
28
+ # "BboxHead": BboxHead,
29
+ # "LandmarkHead": LandmarkHead,
30
+ # "ClassHead": ClassHead}
31
  )
32
 
33
  arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFace")