Piano-Classification

Classify piano sound quality by fine-tuned pre-trained CNN models.

Requirements

conda create -n py311 python=3.11 -y
conda activate py311
pip install -r requirements.txt

Usage

Maintenance

git clone git@github.com:monetjoe/Piano-Classification.git
cd Piano-Classification

Train

Assign a backbone(take squeezenet1_1 as an example) after --model to start training:

python train.py --model squeezenet1_1 --fullfinetune True --fl True

--fullfinetune True means full finetune, False means linear probing
--fl True means using focal loss

Supported backbones

Mirror 1	Mirror 2

Plot results

After finishing the training, use the below command to plot the latest results:

python plot.py

Results

A demo result of SqueezeNet fine-tuning:

Results	Plots
Loss curve
Training and validation accuracy
Confusion matrix

Cite

@inproceedings{Zhou2023AHE,
  author    = {Monan Zhou and Shangda Wu and Shaohua Ji and Zijin Li and Wei Li},
  title     = {A Holistic Evaluation of Piano Sound Quality},
  booktitle = {Proceedings of the 10th Conference on Sound and Music Technology (CSMT)},
  year      = {2023},
  publisher = {Springer Singapore},
  address   = {Singapore}
}