diff --git a/README.md b/README.md index c97aea3..c192d2c 100644 --- a/README.md +++ b/README.md @@ -8,7 +8,13 @@ install requirements: ```bash pip install -r requirements.txt ``` - +This code also requires the Synchronized-BatchNorm-PyTorch rep. +``` +cd models/networks/ +git clone https://github.com/vacancy/Synchronized-BatchNorm-PyTorch +cp -rf Synchronized-BatchNorm-PyTorch/sync_batchnorm . +cd ../../ +``` ## Code Structure The code is heavily based on the [official implementation](https://github.com/NVlabs/SPADE) of [SPADE](https://arxiv.org/pdf/1903.07291.pdf), and therefore has the saome structure: - `train.py`, `test.py`: the entry point for training and testing. @@ -93,4 +99,4 @@ Attribution-NonCommercial-ShareAlike 4.0 International (see file). booktitle={Computer Vision and Pattern Recognition (CVPR)}, year={2021} } -``` \ No newline at end of file +```