tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Refactor: dataset_root and assets_root in configs/paths.yml

Browse Source
This commit is contained in:
litagin02
2024-01-02 22:51:17 +09:00
parent a1069c73d2
commit 98ee6f4c69
12 changed files with 209 additions and 162 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
webui_dataset.py
View File

@@ -1,15 +1,22 @@
import os
import gradio as gr
import yaml
from common.log import logger
from common.subprocess_utils import run_script_with_log, second_elem_of
from common.subprocess_utils import run_script_with_log
# Get path settings
with open(os.path.join("configs", "paths.yml"), "r", encoding="utf-8") as f:
path_config: dict[str, str] = yaml.safe_load(f.read())
dataset_root = path_config["dataset_root"]
# assets_root = path_config["assets_root"]
def do_slice(model_name, normalize):
logger.info("Start slicing...")
input_dir = "inputs"
output_dir = os.path.join("Data", model_name, "raw")
output_dir = os.path.join(dataset_root, model_name, "raw")
cmd = [
"slice.py",
"--input_dir",
@@ -26,8 +33,8 @@ def do_slice(model_name, normalize):
def do_transcribe(model_name):
input_dir = os.path.join("Data", model_name, "raw")
output_file = os.path.join("Data", model_name, "esd.list")
input_dir = os.path.join(dataset_root, model_name, "raw")
output_file = os.path.join(dataset_root, model_name, "esd.list")
result = run_script_with_log(
[
"transcribe.py",