From 89b9a2c68006225f0904cfb6b2a9d761b2f5e474 Mon Sep 17 00:00:00 2001
From: Guochen Yu <yuguochen@cuc.edu.cn>
Date: Mon, 24 Jan 2022 19:16:31 +0800
Subject: [PATCH] Add files via upload

---
 json_extract.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 json_extract.py

diff --git a/json_extract.py b/json_extract.py
new file mode 100644
index 0000000..3e654c5
--- /dev/null
+++ b/json_extract.py
@@ -0,0 +1,30 @@
+import json
+import os
+import numpy as np
+import random
+
+def json_ext(path, flag):
+    if flag is 'train':
+        dataset_path = os.path.join(path, 'train', 'mix')# read Train mixed speech name
+        json_path = os.path.join(path, 'Json', 'train')
+        os.makedirs(json_path, exist_ok=True)
+    else:
+        dataset_path = os.path.join(path, 'cv', 'mix')# read CV mixed speech name 
+        json_path = os.path.join(path, 'Json', 'cv')
+        os.makedirs(json_path, exist_ok=True)
+    data_dir = os.listdir(dataset_path)
+    file_num = len(data_dir)
+    random.shuffle(data_dir)
+    data_list = []
+
+    for i in range(file_num):
+        file_name = data_dir[i]
+        file_name = os.path.splitext(file_name)[0]
+        data_list.append(file_name)
+
+    with open(os.path.join(json_path, 'files.json'), 'w') as f :
+        json.dump(data_list, f, indent=4)
+
+file_path = 'xxx'
+json_ext(file_path, flag='train')
+json_ext(file_path, flag='cv')
\ No newline at end of file