speech_recognition/esp-sr
1
0
Fork 0
mirror of https://github.com/espressif/esp-sr.git synced 2025-09-15 15:28:44 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

bugfix(model): Fix the bug of moving model on Windows

Browse Source
This commit is contained in:
Wang Wang Wang 2021-12-02 14:23:09 +08:00
parent 509cf82658
commit ed1e9d451e
1 changed files with 7 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
model/movemodel.py
View File

@ -1,7 +1,7 @@
import io import io
import os import os
import argparse import argparse
import shutil
def calculate_total_size(folder_path): def calculate_total_size(folder_path):
total_size = 0 total_size = 0
for file_name in os.listdir(folder_path): for file_name in os.listdir(folder_path):
@ -69,22 +69,18 @@ elif "CONFIG_SR_MN_EN_MULTINET5_SINGLE_RECOGNITION" in MN_STRING:
else: else:
print('choose no multinet mode') print('choose no multinet mode')
if wakenet_model != 'null':
wakenet_model = model_path + '/wakenet_model/' + wakenet_model
if multinet_model != 'null':
multinet_model = model_path + '/multinet_model/' + multinet_model
print(wakenet_model) print(wakenet_model)
print(multinet_model) print(multinet_model)
target_model = model_path + '/target' target_model = model_path + '/target'
if not os.path.exists(target_model):
os.makedirs(target_model) shutil.rmtree(target_model)
os.system("rm -rf %s/*" % target_model) os.makedirs(target_model)
if wakenet_model != 'null': if wakenet_model != 'null':
os.system("cp -rf %s %s" % (wakenet_model, target_model)) shutil.copytree(model_path + '/wakenet_model/' + wakenet_model, target_model+'/'+wakenet_model)
if multinet_model != 'null': if multinet_model != 'null':
os.system("cp -rf %s %s" % (multinet_model, target_model)) shutil.copytree(model_path + '/multinet_model/' + multinet_model, target_model+'/'+multinet_model)
# os.system("cp %s %s" % (wakenet_model+'/_MODEL_INFO_', target_model)) # os.system("cp %s %s" % (wakenet_model+'/_MODEL_INFO_', target_model))
total_size = calculate_total_size(target_model) total_size = calculate_total_size(target_model)