yufeng0528 7 months ago
parent
commit
ac73267e5e
2 changed files with 40 additions and 4 deletions
  1. 2 1
      draw/requirement.txt
  2. 38 3
      draw/web_index.py

+ 2 - 1
draw/requirement.txt

@@ -1,4 +1,5 @@
1 1
 opencv-python-headless
2 2
 tensorflow
3 3
 opencv-python
4
-mtcnn
4
+mtcnn
5
+traceback2

+ 38 - 3
draw/web_index.py

@@ -6,6 +6,31 @@ import MTCNN_person
6 6
 
7 7
 app = Flask(__name__)
8 8
 
9
+import datetime
10
+import logging
11
+def get_today_str(plusDay=0):
12
+    today = datetime.datetime.now()
13
+    today = today + datetime.timedelta(days=plusDay)
14
+    return today.strftime('%Y-%m-%d')
15
+
16
+today = get_today_str()
17
+def logger(log_obj, log_file="/data/logs/face-info-" + today.replace('-', '') +".log"):
18
+    logger_obj = logging.getLogger(log_obj)
19
+    logger_obj.setLevel(logging.INFO)
20
+
21
+    file_handle = logging.FileHandler(log_file)
22
+    file_handle.setLevel(logging.INFO)
23
+
24
+    formatter = logging.Formatter(
25
+        '%(asctime)s - %(name)s - %(levelname)s - %(message)s')
26
+
27
+    file_handle.setFormatter(formatter)
28
+
29
+    logger_obj.addHandler(file_handle)
30
+
31
+    return logger_obj
32
+_logger = logger(__name__)
33
+
9 34
 @app.route('/')
10 35
 def hello_world():
11 36
     return 'Hello World!'
@@ -42,16 +67,26 @@ def download():
42 67
 def face():
43 68
     info = request.args.get('url')
44 69
     time1 = int(round(time.time() * 1000))
45
-    print('info=%s' % info)
70
+    _logger.info('info=%s' % info)
46 71
     name = do_download(info)
47 72
     time2 = int(round(time.time() * 1000))
48 73
     faces = MTCNN_person.detector(dir + name)
49 74
     time3 = int(round(time.time() * 1000))
50
-    print('over %s from %s' % (time3 - time1, time1))
75
+    _logger.info('[%s] over %s from %s' % (name, time3 - time1, time1))
51 76
     return json.dumps(faces)
52 77
 
53 78
 
79
+from gevent import pywsgi
80
+import traceback
54 81
 if __name__ == '__main__':
55
-    app.run(port=5000)
82
+    try:
83
+        # app.run(host='192.168.0.220', port=5000)
84
+        server = pywsgi.WSGIServer(('0.0.0.0', 5000), app)
85
+        server.serve_forever()
86
+    except Exception as e:
87
+        traceback.print_exc(file=open('error.txt', 'a+'))
88
+        print(e)
89
+    # server = pywsgi.WSGIServer(('0.0.0.0', 5000), app)
90
+    # server.serve_forever()
56 91
     # do_download(url='https://img.bbztx.com/article/live150178/team217182/20230913192631_tmp_9fbaaecf98fd1a3e91a6571f744f4b34511300d6c5a5b358.jpg?x-oss-process=image/resize,h_1000,m_lfit')
57 92
     # MTCNN_person.detector('C:\\Users\\yufen\\Pictures\\face\\20230913192631_tmp_9fbaaecf98fd1a3e91a6571f744f4b34511300d6c5a5b358.jpg')