All-In-One/script.py at master · samimideksa/All-In-One · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
import sys
sys.path.insert(0, 'grpc/')

from concurrent import futures
import time
import grpc
from keras.models import model_from_json
from keras.models import Model
from scipy.io import loadmat
import os
import cv2
import numpy as np
import pandas as pd
import dlib
import all_in_one_pb2
import all_in_one_pb2_grpc
from demo import load_model, selective_search_demo
from demo.main import images_demo, video_demo, process_video, webcam_demo, video_demo
import sys
_ONE_DAY_IN_SECONDS = 60 * 60 * 24

class Greeter(all_in_one_pb2_grpc.AllInOneServicer):
    def SayHello(self, request, context):
        return all_in_one_pb2.AllInOneResponse(message="Hello, %s" % request.name)

    def AllInOne(self, request, context):
        return all_in_one_pb2.AllInOneResponse()

def serve():
    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
    all_in_one_pb2_grpc.add_AllInOneServicer_to_server(Greeter(), server)
    server.add_insecure_port('[::]:50051')
    server.start()
    model  = load_model("/home/samuel/projects/All-In-One/allinonemodels/allinone.json","/home/samuel/projects/All-In-One/allinonemodels/freeze2.h5",["age_estimation","smile", "gender_probablity"])
    model.summary()
    detector = dlib.get_frontal_face_detector()
    #image demo
    images_demo(model,"/home/samuel/projects/All-In-One/grpc/test/test_images/",detector)
    #video demo
    #webcam_demo(model,detector)
    video_demo(model,"/home/samuel/projects/All-In-One/grpc/test/test_videos/video_demo3.mp4",detector)
    try:
        while True:
            time.sleep(_ONE_DAY_IN_SECONDS)
    except KeyboardInterrupt:
        server.stop(0)

if __name__ == '__main__':
    model  = load_model("/home/samuel/projects/All-In-One/allinonemodels/allinone.json","/home/samuel/projects/All-In-One/allinonemodels/freeze2.h5",["age_estimation","smile", "gender_probablity"])
    model.summary()
    detector = dlib.get_frontal_face_detector()
    #image demo
    images_demo(model,"/home/samuel/projects/All-In-One/grpc/test/test_images/",detector)
    #video demo
    #webcam_demo(model,detector)
    video_demo(model,"/home/samuel/projects/All-In-One/grpc/test/test_videos/video_demo3.mp4",detector)
    serve()