Anuj-Panthri commited on
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 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],num_anchors,self.last_item, self.num_anchors,self.last_item)
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):