-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
87 lines (76 loc) · 3.58 KB
/
main.py
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#-*-coding:utf-8-*-
'''
DpCas-Light
|||| ||||| |||| || |||||||
|| || || || || || |||| || ||
|| || || || || || || || ||
|| || || || || ||====|| ||||||
|| || ||||| || || ||======|| ||
|| || || || || || || || ||
|||| || |||| || || |||||||
/-------------------- APP_X --------------------/
'''
# date:2020-10-19.7.23.24
# Author: Eric.Lee
# function: main
import os
import argparse
import warnings
warnings.filterwarnings("ignore")
import sys
sys.path.append("./components/") # 添加模型组件路径
def demo_logo():
print("\n/*********************************/")
print("/---------------------------------/\n")
print(" WELCOME : DpCas-Light ")
print(" << APP_X >> ")
print(" Copyright 2021 Eric.Lee2021 ")
print(" Apache License 2.0 ")
print("\n/---------------------------------/")
print("/*********************************/\n")
if __name__ == '__main__':
demo_logo()
parser = argparse.ArgumentParser(description= " DpCas : << Deep Learning Componentized Application System >> ")
parser.add_argument('-app', type=int, default = 0,
help = "handpose_x:0, gesture:1 ,wyw2s:2, face_bioassay:3, video_ana:4, face_pay:5, drive:6") # 设置 App Example
app_dict = {
0:"handpose_x",
1:"gesture",
2:"wyw2s",
3:"face_bioassay",
4:"video_ana",
5:"face_pay",
6:"drive"}
args = parser.parse_args()# 解析添加参数
APP_P = app_dict[args.app]
if APP_P == "handpose_x": # 手势识别
from applications.handpose_local_app import main_handpose_x #加载 handpose 应用
cfg_file = "./lib/hand_lib/cfg/handpose.cfg"
main_handpose_x(cfg_file)#加载 handpose 应用
elif APP_P == "gesture": # 手势识别
from applications.gesture_local_app import main_gesture_x #加载 gesture 应用
cfg_file = "./lib/gesture_lib/cfg/handpose.cfg"
main_gesture_x(cfg_file)#加载 handpose 应用
elif APP_P == "wyw2s": # 基于人脸识别的视频剪辑
from applications.wyw2s_local_app import main_wyw2s #加载 who you want 2 see 应用
cfg_file = "./lib/wyw2s_lib/cfg/wyw2s.cfg"
main_wyw2s(video_path = "./video/rw_11.mp4",cfg_file = cfg_file)#加载 who you want 2 see 应用
elif APP_P == "face_bioassay":
from applications.face_bioassay_local_app import main_face_bioassay #face_bioassay 应用
cfg_file = "./lib/face_bioassay_lib/cfg/face_bioassay.cfg"
video_path = "./video/face2.mp4"
main_face_bioassay(video_path = video_path,cfg_file = cfg_file)#加载 face_bioassay 应用
# elif APP_P == "video_ana":
# from applications.VideoAnalysis_app import main_VideoAnalysis #加载 video_analysis 应用
# main_VideoAnalysis(video_path = "./video/f3.mp4")#加载 video_analysis 应用
#
# elif APP_P == "face_pay":
# cfg_file = "./lib/facepay_lib/cfg/facepay.cfg"
# from applications.FacePay_local_app import main_facePay #加载 face pay 应用
# main_facePay(video_path = 0,cfg_file = cfg_file) # 加载 face pay 应用
#
# elif APP_P == "drive":
# from applications.DangerousDriveWarning_local_app import main_DangerousDriveWarning #加载 危险驾驶预警 应用
# cfg_file = "./lib/dfmonitor_lib/cfg/dfm.cfg"
# main_DangerousDriveWarning(video_path = "./video/drive1.mp4",cfg_file = cfg_file)
print(" well done ~")