depot/third_party/nixpkgs/pkgs/development/python-modules/piano-transcription-inference/default.nix
Default email 87f9c27ba9 Project import generated by Copybara.
GitOrigin-RevId: fe2ecaf706a5907b5e54d979fbde4924d84b65fc
2023-04-12 14:48:02 +02:00

79 lines
2.4 KiB
Nix

{ 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 ];
};
}