Add files via upload

VTTCAU · Apr 28, 2020 · 11c9c2f · 11c9c2f
1 parent 9a9e037
commit 11c9c2f
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/Face_recog/train_main.py b/Face_recog/train_main.py
@@ -0,0 +1,25 @@
+# Face Recognition
+from config import get_config
+from Learner import face_learner
+import argparse
+
+# Person Detector
+from Yolo_v2_pytorch.train_yolo import get_args as get_fd_args
+from Yolo_v2_pytorch.train_yolo import train as fd_train
+
+# TRACKING(FACE RECOGNITION) - not completed
+# ========================================================
+# learner = face_learner(conf)
+# learner.train(conf, args.epochs)
+# ========================================================
+
+# PERSON DETECTOR
+# ========================================================
+fd_args = get_fd_args()
+fd_train(fd_args)
+# ========================================================
+
+# MAKE YOUR LOADER IN YOUR TRAIN CLASS
+# ========================================================
+
+# ========================================================