An open source implementation of the multiple time-step loss functions for peak-picking proposed in [1].
A cnnSA3F5 model [1,2] is utilised with CE, MI, MMD and WMD loss functions for automatic drum transcription using data from [3].
This code is published under the BSD license which allows redistribution and modification as long as the copyright and disclaimers are contained.
The example data is taken from the MDB Drums dataset [3] which is published under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.
The full license information can be found on the license page.
• numpy
• tensorflow(1.8.0)
• matplotlib