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
|
{
stdenv,
lib,
buildPythonPackage,
fetchPypi,
fetchpatch,
fetchurl,
librosa,
matplotlib,
mido,
torch,
torchlibrosa,
}:
buildPythonPackage rec {
pname = "piano-transcription-inference";
version = "0.0.5";
format = "setuptools";
src = fetchPypi {
inherit pname version;
hash = "sha256-nbhuSkXuWrekFxwdNHaspuag+3K1cKwq90IpATBpWPY=";
};
checkpoint = fetchurl {
name = "piano-transcription-inference.pth";
# The download url can be found in
# https://github.com/qiuqiangkong/piano_transcription_inference/blob/master/piano_transcription_inference/inference.py
url = "https://zenodo.org/record/4034264/files/CRNN_note_F1%3D0.9677_pedal_F1%3D0.9186.pth?download=1";
hash = "sha256-w/qXMHJb9Kdi8cFLyAzVmG6s2gGwJvWkolJc1geHYUE=";
};
propagatedBuildInputs = [
librosa
matplotlib
mido
torch
torchlibrosa
];
patches = [
# Fix run against librosa 0.10.0
# https://github.com/qiuqiangkong/piano_transcription_inference/pull/14
(fetchpatch {
url = "https://github.com/qiuqiangkong/piano_transcription_inference/commit/b2d448916be771cd228f709c23c474942008e3e8.patch";
hash = "sha256-8O4VtFij//k3fhcbMRz4J8Iz4AdOPLkuk3UTxuCSy8U=";
})
(fetchpatch {
url = "https://github.com/qiuqiangkong/piano_transcription_inference/commit/61443632dc5ea69a072612b6fa3f7da62c96b72c.patch";
hash = "sha256-I9+Civ95BnPUX0WQhTU/pGQruF5ctIgkIdxCK+xO3PE=";
})
];
postPatch = ''
substituteInPlace piano_transcription_inference/inference.py --replace \
"checkpoint_path='{}/piano_transcription_inference_data/note_F1=0.9677_pedal_F1=0.9186.pth'.format(str(Path.home()))" \
"checkpoint_path='$out/share/checkpoint.pth'"
'';
postInstall = ''
mkdir "$out/share"
ln -s "${checkpoint}" "$out/share/checkpoint.pth"
'';
# Project has no tests.
# In order to make pythonImportsCheck work, NUMBA_CACHE_DIR env var need to
# be set to a writable dir (https://github.com/numba/numba/issues/4032#issuecomment-488102702).
# pythonImportsCheck has no pre* hook, use checkPhase to wordaround that.
checkPhase = ''
export NUMBA_CACHE_DIR="$(mktemp -d)"
'';
pythonImportsCheck = [ "piano_transcription_inference" ];
meta = with lib; {
description = "A piano transcription inference package";
homepage = "https://github.com/qiuqiangkong/piano_transcription_inference";
license = licenses.mit;
maintainers = with maintainers; [ azuwis ];
};
}
|
