diff --git a/.gitignore b/.gitignore index 65108e2..9f1be9a 100644 --- a/.gitignore +++ b/.gitignore @@ -9,4 +9,7 @@ __pycache__/ **/.DS_Store # tmp -~* \ No newline at end of file +~* + +# pretrained_models +pretrained_models/ \ No newline at end of file diff --git a/README.md b/README.md index bebc96f..78fd9e9 100644 --- a/README.md +++ b/README.md @@ -60,6 +60,10 @@ python -m pip install torch torchvision torchaudio python -m pip install -e segment_anything python -m pip install -r lama/requirements.txt ``` +On Windows system, we recommend to first install [miniconda](https://docs.conda.io/en/latest/miniconda.html) and +open `Anaconda Powershell Prompt (miniconda3)` as administrator. +Then pip install [./lama_requirements_windows.txt](lama_requirements_windows.txt) instead of +[./lama/requirements.txt](lama%2Frequirements.txt). ### Usage Download the model checkpoints provided in [segment_anything](./segment_anything/README.md) diff --git a/lama_requirements_windows.txt b/lama_requirements_windows.txt new file mode 100644 index 0000000..1bcef6c --- /dev/null +++ b/lama_requirements_windows.txt @@ -0,0 +1,19 @@ +pyyaml +tqdm +numpy +easydict==1.9.0 +scikit-image +scikit-learn +opencv-python +tensorflow +joblib +matplotlib +pandas +albumentations==0.5.2 +hydra-core==1.1.0 +pytorch-lightning==1.2.9 +tabulate +kornia==0.5.0 +webdataset +packaging +wldhx.yadisk-direct