diff --git a/README.md b/README.md index 18b4aca..f92aa76 100644 --- a/README.md +++ b/README.md @@ -36,7 +36,7 @@ The overview of our ZITS. At first, the TSR model is used to restore structures pip install -r requirement.txt git clone https://github.com/NVIDIA/apex cd apex - pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" ./ + pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --no-build-isolation ./ ``` 2. For training, [MST](https://github.com/ewrfcas/MST_inpainting) provide irregular and segmentation masks ([download](https://drive.google.com/drive/folders/1eU6VaTWGdgCXXWueCXilt6oxHdONgUgf?usp=sharing)) with different masking rates. And you should define the mask file list before the training as in [MST](https://github.com/ewrfcas/MST_inpainting). @@ -178,4 +178,4 @@ If you found our program helpful, please consider citing: booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition}, year={2022} } -``` \ No newline at end of file +```