Zalo AI Challenge

Repo phục vụ cho Zalo AI Challenge: Voice Gender Classification

Nhóm tác giả:

• Vũ Anh [email protected]
• Truong Do [email protected]
• Nguyễn Thế Anh [email protected]

Trích chọn đặc trưng

Tập tin âm thanh được chia thành những frames dài 250ms không trùng lặp. Mỗi frame được gán với label tương ứng với tập tin âm thanh (gồm gender và ascent). Từ mỗi frame, trích chọn đặc trưng melspectrogram với 60 bands. Kết quả được cho vào hàm logamplitude. Tập tin âm thanh đầu vào được đầu bởi thư viện librosa. Sau đó, kết quả với đặc trưng delta tương ứng. Kết quả được đưa vào mạng convolutional neural network.

Kiến trúc mạng convolutional neural network

Mô hình được train bởi hàm tối ưu Adam với learning rate = 0.0001, beta_1 = 0.9, beta=0.999

Dự đoán

Mỗi một sample được chia thành mỗi 250 frames, sử dụng phương pháp trích rút đặc trưng như mô tả ở trên, rồi đưa vào mạng CNN. Nhãn của file âm thanh được chọn bởi chiến thuật majority voting.

Hệ thống chạy trên môi trường python 3.6, sử dụng keras và tensorflow. Quá trình trích rút đặc trưng được thực hiện bởi thư viện librosa.

pip install requirements.txt

Để huấn luyện mô hình, chạy script preprocessing.py và train.py

Chú ý: Dữ liệu train gồm có folder train cần đặt vào thư mục data

python preprocessing.py --mode train
python train.py

Để dự đoán, chạy script predict.py

python preprocessing.py --mode test
python predict.py