|
import os |
|
import sys |
|
|
|
|
|
def input_pose(pose_select="front"): |
|
step = 1 |
|
if pose_select == "front": |
|
pose = [[0.0, 0.0, 0.0] for i in range(0, 10, step)] |
|
elif pose_select == "left_right_shaking": |
|
pose = [[-i, 0.0, 0.0] for i in range(0, 20, step)] |
|
pose += [[i - 20.0, 0.0, 0.0] for i in range(0, 40, step)] |
|
pose += [[20.0 - i, 0.0, 0.0] for i in range(0, 20, step)] |
|
pose = pose + pose |
|
pose = pose + pose |
|
pose = pose + pose |
|
|
|
else: |
|
raise ValueError("pose_select Error") |
|
|
|
return pose |
|
|
|
|
|
EMOTIONS = ['angry', 'disgust', 'fear', 'happy', 'sad', 'surprise', 'neutral'] |
|
def input_emotion(emotion_select="neutral"): |
|
sacle_factor = 2 |
|
if emotion_select == "neutral": |
|
emotion = [[0.0,0.0,0.0,0.0,0.0,0.0,1.0] for _ in range(2)] |
|
elif emotion_select == "happy": |
|
emotion = [[0.0,0.0,0.0,1.0,0.0,0.0,0.0] for _ in range(2)] |
|
elif emotion_select == "angry": |
|
emotion = [[1.0,0.0,0.0,0.0,0.0,0.0,0.0] for _ in range(2)] |
|
elif emotion_select == "surprised": |
|
emotion = [[0.0,0.0,0.0,0.0,0.0,1.0,0.0] for _ in range(2)] |
|
else: |
|
raise ValueError("emotion_select Error") |
|
|
|
return emotion * sacle_factor |
|
|
|
|
|
def input_blink(blink_select="yes"): |
|
if blink_select == "yes": |
|
blink = [[1.0], [1.0], [1.0], [1.0], [1.0], [1.0], [1.0], [0.8], [0.6], [0.0], [0.0], [1.0]] |
|
blink = blink + blink + blink |
|
else: |
|
blink = [[1.0] for _ in range(2)] |
|
return blink |
|
|
|
|
