139 lines
2.5 KiB
Nix
139 lines
2.5 KiB
Nix
# WARNING: This file was automatically generated. You should avoid editing it.
|
|
# If you run pynixify again, the file will be either overwritten or
|
|
# deleted, and you will lose the changes you made to it.
|
|
|
|
{ addict
|
|
, basicsr
|
|
, stdenv
|
|
, future
|
|
, gdown
|
|
, lib
|
|
, lmdb
|
|
, lpips
|
|
, numpy
|
|
, opencv-python
|
|
, pillow
|
|
, pyyaml
|
|
, requests
|
|
, scikit-image
|
|
, scipy
|
|
, fetchFromGitHub
|
|
, buildPythonPackage
|
|
, torch
|
|
, cython
|
|
, torchvision
|
|
, tqdm
|
|
, wget
|
|
, yapf
|
|
, pythonRelaxDepsHook
|
|
}:
|
|
|
|
let
|
|
nestedBasicsr = buildPythonPackage
|
|
{
|
|
pname = "basicsr";
|
|
version = "0.1.2";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "sczhou";
|
|
repo = "CodeFormer";
|
|
rev = "c5b4593074ba6214284d6acd5f1719b6c5d739af";
|
|
sha256 = "sha256-JyyJe+VBeNK5rRaPJ4jYdKZqLnRfayHWkTwFNrSfseY=";
|
|
};
|
|
|
|
preBuild = ''
|
|
ln -s basicsr/setup.py setup.py
|
|
'';
|
|
|
|
postPatch = ''
|
|
substituteInPlace requirements.txt \
|
|
--replace opencv-python "" \
|
|
--replace tb-nightly ""
|
|
'';
|
|
|
|
|
|
nativeBuildInputs = [ cython ];
|
|
|
|
propagatedBuildInputs = [
|
|
pillow
|
|
pyyaml
|
|
addict
|
|
future
|
|
gdown
|
|
lmdb
|
|
lpips
|
|
numpy
|
|
opencv-python
|
|
requests
|
|
scikit-image
|
|
scipy
|
|
torch
|
|
torchvision
|
|
tqdm
|
|
wget
|
|
yapf
|
|
];
|
|
|
|
doCheck = false;
|
|
|
|
meta = with lib; {
|
|
description =
|
|
"Towards Robust Blind Face Restoration with Codebook Lookup Transformer (NeurIPS 2022)";
|
|
homepage = "None";
|
|
};
|
|
};
|
|
in
|
|
stdenv.mkDerivation rec {
|
|
pname = "codeformer";
|
|
version = "0.1.2";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "sczhou";
|
|
repo = "CodeFormer";
|
|
rev = "c5b4593074ba6214284d6acd5f1719b6c5d739af";
|
|
sha256 = "sha256-JyyJe+VBeNK5rRaPJ4jYdKZqLnRfayHWkTwFNrSfseY=";
|
|
};
|
|
|
|
|
|
buildPhase = ''
|
|
mkdir -p $out/lib/python3.10/site-packages
|
|
cp -R ./ $out/lib/python3.10/site-packages/codeformer/
|
|
cd $out/lib/python3.10/site-packages/codeformer/
|
|
cp -R ${nestedBasicsr}/lib/python3.10/site-packages/basicsr ./basicsr/
|
|
'';
|
|
|
|
dontInstall = true;
|
|
|
|
patches = [ ./root_dir.patch ];
|
|
|
|
|
|
propagatedBuildInputs = [
|
|
pillow
|
|
pyyaml
|
|
addict
|
|
future
|
|
gdown
|
|
lmdb
|
|
lpips
|
|
numpy
|
|
opencv-python
|
|
requests
|
|
scikit-image
|
|
scipy
|
|
torch
|
|
torchvision
|
|
tqdm
|
|
wget
|
|
yapf
|
|
];
|
|
|
|
# TODO FIXME
|
|
doCheck = false;
|
|
|
|
meta = with lib; {
|
|
description =
|
|
"Towards Robust Blind Face Restoration with Codebook Lookup Transformer (NeurIPS 2022)";
|
|
homepage = "None";
|
|
};
|
|
}
|