in svoice/data/data.py [0:0]
def __init__(self, json_dir):
mix_json = os.path.join(json_dir, 'mix.json')
s_jsons = list()
s_infos = list()
sets_re = re.compile(r's[0-9].json')
for s in os.listdir(json_dir):
if sets_re.search(s):
s_jsons.append(os.path.join(json_dir, s))
with open(mix_json, 'r') as f:
mix_infos = json.load(f)
for s_json in s_jsons:
with open(s_json, 'r') as f:
s_infos.append(json.load(f))
self.mix_set = Audioset(sort(mix_infos))
self.sets = list()
for s_info in s_infos:
self.sets.append(Audioset(sort(s_info)))
for s in self.sets:
assert len(s) == len(self.mix_set)