.

app.py

@@ -20,14 +20,14 @@ opt = SwapOptions().parse()
 
 retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/RetinaFace")
 
-from retina_model import *
-
-RetinaFace = load_model("retina_model/RetinaFace-Res50.h5"
-                        # custom_objects={"FPN": FPN,
-                        #                 "SSH": SSH,
-                        #                 "BboxHead": BboxHead,
-                        #                 "LandmarkHead": LandmarkHead,
-                        #                 "ClassHead": ClassHead}
+from retina_model.models import *
+
+RetinaFace = load_model("retina_model/RetinaFace-Res50.h5",
+                        custom_objects={"FPN": FPN,
+                                        "SSH": SSH,
+                                        "BboxHead": BboxHead,
+                                        "LandmarkHead": LandmarkHead,
+                                        "ClassHead": ClassHead}
                         )
 
 arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFace")