wbrawner/reone
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Rename examine_lip to inspect_lip, improve output

Browse source
This commit is contained in:
Vsevolod Kremianskii 2021-04-22 18:51:55 +07:00
parent 25253405ce
commit 2a7c19eed9
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
scripts/examine_lip.py → scripts/inspect_lip.py
View file

@ -26,7 +26,7 @@ if not os.path.exists(extract_dir):
def _describe_shape(shape):
shapes = [
phonemes = [
"EE",
"EH",
"SCHWA",
@ -44,7 +44,12 @@ def _describe_shape(shape):
"L/R",
"K/G"]
return shapes[shape]
return phonemes[shape]
def normalize_text(text):
result = "".join(filter(lambda s: str.isalpha(s) or s == " ", text)) # strip string of everything but letters and spaces
return result.upper()
def examine_lip(extract_dir, strref):
@ -62,12 +67,13 @@ def examine_lip(extract_dir, strref):
if os.path.exists(lip_path):
with open(lip_path) as lip:
lip_json = json.load(lip)
print(text)
print(" ".join([_describe_shape(int(x["shape"])) for x in lip_json["keyframes"]]))
print("Text (original): " + text)
print("Text (normalized): " + normalize_text(text))
print("Phonemes: " + " ".join([_describe_shape(int(x["shape"])) for x in lip_json["keyframes"]]))
if len(sys.argv) > 1:
strref = int(sys.argv[1])
examine_lip(extract_dir, strref)
else:
print("Usage: python examine_lip.py STRREF")
print("Usage: python inspect_lip.py STRREF")