Skip to content

Commit

Permalink
poster upload
Browse files Browse the repository at this point in the history
  • Loading branch information
edgarschnfld committed Apr 30, 2021
1 parent 051e6bf commit 4eba274
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 3 deletions.
16 changes: 13 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -44,9 +44,6 @@ Training on 4 NVIDIA Tesla V100 (32GB) is recommended.

To test a trained model, execute the testing scripts in the ```scripts``` folder. The ```--name``` parameter should correspond to the experiment name that you want to test, and the ```--checkpoints_dir``` should the folder where the experiment is saved (default: ```./checkpoints```). These scripts will generate images from a pretrained model in ```./results/name/```.

## Video Summary
[![video summary](youtube.png)](https://www.youtube.com/watch?v=vUm6vurIwyM)

## Measuring FID

The FID is computed on the fly during training, using the popular PyTorch FID implementation from https://github.com/mseitzer/pytorch-fid. At the beginning of training, the inception moments of the real images are computed before the actual training loop starts. How frequently the FID should be evaluated is controlled via the parameter ```--freq_fid```, which is set to 5000 steps by default. The inception net that is used for FID computation automatically downloads a pre-trained inception net checkpoint. If that automatic download fails, for instance because your server has restricted internet access, get the checkpoint named ```pt_inception-2015-12-05-6726825d.pth``` from [here](https://www.dropbox.com/sh/nf6of02pyk84zjg/AAC8hnnj0T_MAiPx3tzdAyiWa?dl=0) and place it in ```/utils/fid_folder/```. In this case, do not forget to replace ```load_state_dict_from_url``` function accordingly.
Expand All @@ -69,6 +66,19 @@ python test.py --dataset_mode ade20k --name oasis_ade20k_pretrained \
This script will create a folder named ```./results``` in which the resulting images are saved.

If you want to continue training from this checkpoint, use ```train.py``` with the same ```--name``` parameter and add ```--continue_train --which_iter best```.

# Additional information

## Poster

<p align="center">
<img src="iclr2021_oasis_poster.png" >
</p>


## Video Summary
[![video summary](youtube.png)](https://www.youtube.com/watch?v=vUm6vurIwyM)

## Citation
If you use this work please cite
```
Expand Down
Binary file added iclr2021_oasis_poster.png
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit 4eba274

Please sign in to comment.