Spaces:
Sleeping
Sleeping
Commit
·
40c8935
1
Parent(s):
dc31c84
trying new face detection model (trained on mid size faces only(scale factor 6-8))
Browse files- app/__init__.py +2 -1
- face_detection/Models/BestMap/anchors.txt +5 -0
- face_detection/Models/BestMap/config.py +2 -0
- face_detection/Models/BestMap/model.h5 +3 -0
- face_detection/Models/Final/anchors.txt +5 -0
- face_detection/Models/Final/config.py +2 -0
- face_detection/Models/Final/model.h5 +3 -0
- face_detection/create_load_model.py +1 -1
app/__init__.py
CHANGED
|
@@ -10,7 +10,8 @@ from face_recognition import helper as fr_helper
|
|
| 10 |
|
| 11 |
|
| 12 |
# face_detector=fd.face_detection("face_detection/Models/v1")
|
| 13 |
-
face_detector=fd.face_detection("face_detection/Models/mobilenet")
|
|
|
|
| 14 |
face_detector.square_preprocessing=fd.square_pad()
|
| 15 |
# face_recognizer=fr.face_recognition("face_recognition/Models/v1")
|
| 16 |
# face_recognizer=fr.face_recognition("face_recognition/Models/mobilenet_basic_lfw")
|
|
|
|
| 10 |
|
| 11 |
|
| 12 |
# face_detector=fd.face_detection("face_detection/Models/v1")
|
| 13 |
+
# face_detector=fd.face_detection("face_detection/Models/mobilenet")
|
| 14 |
+
face_detector=fd.face_detection("face_detection/Models/BestMap")
|
| 15 |
face_detector.square_preprocessing=fd.square_pad()
|
| 16 |
# face_recognizer=fr.face_recognition("face_recognition/Models/v1")
|
| 17 |
# face_recognizer=fr.face_recognition("face_recognition/Models/mobilenet_basic_lfw")
|
face_detection/Models/BestMap/anchors.txt
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
1.301617145538330078e+00 1.570287585258483887e+00
|
| 2 |
+
2.205962896347045898e+00 2.635638713836669922e+00
|
| 3 |
+
2.868087053298950195e+00 3.392648935317993164e+00
|
| 4 |
+
3.687522888183593750e+00 4.747292041778564453e+00
|
| 5 |
+
6.475809812545776367e-01 7.812500000000000000e-01
|
face_detection/Models/BestMap/config.py
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
p_thres=0.6
|
| 2 |
+
nms_thres=0.3
|
face_detection/Models/BestMap/model.h5
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:511ad38f453eb23d63105801636e250aea5b13c1448fe8837c43900524a89c2f
|
| 3 |
+
size 39289720
|
face_detection/Models/Final/anchors.txt
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
1.301617145538330078e+00 1.570287585258483887e+00
|
| 2 |
+
2.205962896347045898e+00 2.635638713836669922e+00
|
| 3 |
+
2.868087053298950195e+00 3.392648935317993164e+00
|
| 4 |
+
3.687522888183593750e+00 4.747292041778564453e+00
|
| 5 |
+
6.475809812545776367e-01 7.812500000000000000e-01
|
face_detection/Models/Final/config.py
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
p_thres=0.6
|
| 2 |
+
nms_thres=0.3
|
face_detection/Models/Final/model.h5
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d1d9c69962063528eb78b630c6552cc0605804b900d9d59b9e875cb412f149ed
|
| 3 |
+
size 39289720
|
face_detection/create_load_model.py
CHANGED
|
@@ -18,7 +18,7 @@ class yolo_reshape(tf.keras.layers.Layer):
|
|
| 18 |
return tf.reshape(output_layer,[shape[0],shape[1],shape[2],self.num_anchors,self.last_item])
|
| 19 |
|
| 20 |
def compute_output_shape(self, input_shape):
|
| 21 |
-
return (input_shape[0],input_shape[1],input_shape[2],
|
| 22 |
|
| 23 |
|
| 24 |
def get_config(self):
|
|
|
|
| 18 |
return tf.reshape(output_layer,[shape[0],shape[1],shape[2],self.num_anchors,self.last_item])
|
| 19 |
|
| 20 |
def compute_output_shape(self, input_shape):
|
| 21 |
+
return (input_shape[0],input_shape[1],input_shape[2],self.num_anchors,self.last_item)
|
| 22 |
|
| 23 |
|
| 24 |
def get_config(self):
|