Browse Source

update reconition face

zangruirui 7 years ago
parent
commit
d0e40a81f5
2 changed files with 3 additions and 3 deletions
  1. 1 1
      README.md
  2. 2 2
      face_recognition_api.py

+ 1 - 1
README.md

@@ -18,7 +18,7 @@
 百度网盘链接 : 链接:http://pan.baidu.com/s/1i4YhAdB 密码:avbl
 
 ### 如何使用?
-首先在服务器上 执行 ###### python   face_recognition_api.py  
+首先在服务器上 执行 #### python   face_recognition_api.py  
 访问地址是XXXXXX:8088  这个可以配置 (代码中有注释)
 模拟post请求,如图所示
 图中依次是插入、查询时的场景

+ 2 - 2
face_recognition_api.py

@@ -134,7 +134,7 @@ with tf.Graph().as_default():
         images = image_array_align_data(img, image_path, pnet, rnet, onet)
 
         #判断如果如图没有检测到人脸则直接返回
-        if images.shape[0] == 1: return json.dumps({'error': "not found face"})
+        if len(images.shape) < 4: return json.dumps({'error': "not found face"})
 
         feed_dict = {images_placeholder: images, phase_train_placeholder: False}
         emb_array = sess.run(embeddings, feed_dict=feed_dict)
@@ -155,7 +155,7 @@ with tf.Graph().as_default():
                 rdict = {'uid': pic_min_uid[i],
                          'distance': pic_min_scores[i],
                          'pic_name': pic_min_names[i] }
-            result.append(rdict)
+				result.append(rdict)
         print(result)
         return json.dumps(result)