File size: 3,446 Bytes
0d80816
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
# Copyright (c) 2023 Amphion.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

import os
import json
import torchaudio
from glob import glob
from collections import defaultdict

from utils.util import has_existed
from preprocessors import GOLDEN_TEST_SAMPLES


def get_test_songs():
    golden_samples = GOLDEN_TEST_SAMPLES["popcs"]
    # every item is a string
    golden_songs = [s.split("_")[:1] for s in golden_samples]
    # song, eg: 万有引力
    return golden_songs


def popcs_statistics(data_dir):
    songs = []
    songs2utts = defaultdict(list)

    song_infos = glob(data_dir + "/*")

    for song_info in song_infos:
        song_info_split = song_info.split("/")[-1].split("-")[-1]

        songs.append(song_info_split)

        utts = glob(song_info + "/*.wav")

        for utt in utts:
            uid = utt.split("/")[-1].split("_")[0]
            songs2utts[song_info_split].append(uid)

    unique_songs = list(set(songs))
    unique_songs.sort()

    print(
        "popcs: {} utterances ({} unique songs)".format(len(songs), len(unique_songs))
    )
    print("Songs: \n{}".format("\t".join(unique_songs)))
    return songs2utts


def main(output_path, dataset_path):
    print("-" * 10)
    print("Preparing test samples for popcs...\n")

    save_dir = os.path.join(output_path, "popcs")
    train_output_file = os.path.join(save_dir, "train.json")
    test_output_file = os.path.join(save_dir, "test.json")
    if has_existed(test_output_file):
        return

    # Load
    popcs_dir = dataset_path

    songs2utts = popcs_statistics(popcs_dir)
    test_songs = get_test_songs()

    # We select songs of standard samples as test songs
    train = []
    test = []

    train_index_count = 0
    test_index_count = 0

    train_total_duration = 0
    test_total_duration = 0

    song_names = list(songs2utts.keys())

    for chosen_song in song_names:
        for chosen_uid in songs2utts[chosen_song]:
            res = {
                "Dataset": "popcs",
                "Singer": "female1",
                "Song": chosen_song,
                "Uid": "{}_{}".format(chosen_song, chosen_uid),
            }
            res["Path"] = "popcs-{}/{}_wf0.wav".format(chosen_song, chosen_uid)
            res["Path"] = os.path.join(popcs_dir, res["Path"])
            assert os.path.exists(res["Path"])

            waveform, sample_rate = torchaudio.load(res["Path"])
            duration = waveform.size(-1) / sample_rate
            res["Duration"] = duration

            if ([chosen_song]) in test_songs:
                res["index"] = test_index_count
                test_total_duration += duration
                test.append(res)
                test_index_count += 1
            else:
                res["index"] = train_index_count
                train_total_duration += duration
                train.append(res)
                train_index_count += 1

    print("#Train = {}, #Test = {}".format(len(train), len(test)))
    print(
        "#Train hours= {}, #Test hours= {}".format(
            train_total_duration / 3600, test_total_duration / 3600
        )
    )

    # Save
    os.makedirs(save_dir, exist_ok=True)
    with open(train_output_file, "w") as f:
        json.dump(train, f, indent=4, ensure_ascii=False)
    with open(test_output_file, "w") as f:
        json.dump(test, f, indent=4, ensure_ascii=False)