Release new trained model and example outputs

daa233 · Jul 11, 2019 · b2d7ded · b2d7ded
1 parent 237df28
commit b2d7ded
Show file tree

Hide file tree

Showing 6 changed files with 2 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -19,11 +19,7 @@ The checkpoints and logs will be saved to `checkpoints`。
 ## Test with the trained model
 By default, it will load the latest saved model in the checkpoints. You can also use `--iter` to choose the saved models by iteration.
 
-Trained PyTorch model: ~~[[Google Drive](https://drive.google.com/drive/folders/1FHlDG94O8sH-sOOpkJBQmxCTicr1gc-C?usp=sharing)] [[Baidu Wangpan](https://pan.baidu.com/s/1c2xWhBIfyz0zttHP6lVR6w)]~~
-
-(Note: The PyTorch model was trained on a bug version whose some channel numbers are wrong. To load the model, you need
-to change the channel numbers according to [here](https://github.com/DAA233/generative-inpainting-pytorch/pull/2/commits/12f49c441edcdf585905f1758f6b5a2f14cb8d86#diff-584f4b3ab12058252e96596e3fd01fffL102).
-The links will be updated after the model is trained from scratch again.)
+Trained PyTorch model: [[Google Drive](https://drive.google.com/open?id=1qbfA5BP9yzdTFFmiOTvYARUYgW1zwBBK)] [[Baidu Wangpan](https://pan.baidu.com/s/17HzpiqMPLIznvCWBfpNVGw)]
 
 ```bash
 python test_single.py \
@@ -45,7 +41,7 @@ python test_tf_model.py \
 
 ## Test results on ImageNet validation set patches
 
-With PyTorch, the model was trained on ImageNet for 50k iterations to converge (about 120h). Here are some test results on the patches from ImageNet validation set.
+With PyTorch, the model was trained on ImageNet for 430k iterations to converge (with batch_size 48, about 150h). Here are some test results on the patches from ImageNet validation set.
 
 | Input | Inpainted |
 |:---:|:---:|

diff --git a/examples/imagenet/imagenet_patches_ILSVRC2012_val_00000827_output.png b/examples/imagenet/imagenet_patches_ILSVRC2012_val_00000827_output.png
diff --git a/examples/imagenet/imagenet_patches_ILSVRC2012_val_00008210_output.png b/examples/imagenet/imagenet_patches_ILSVRC2012_val_00008210_output.png
diff --git a/examples/imagenet/imagenet_patches_ILSVRC2012_val_00022355_output.png b/examples/imagenet/imagenet_patches_ILSVRC2012_val_00022355_output.png
diff --git a/examples/imagenet/imagenet_patches_ILSVRC2012_val_00025892_output.png b/examples/imagenet/imagenet_patches_ILSVRC2012_val_00025892_output.png
diff --git a/examples/imagenet/imagenet_patches_ILSVRC2012_val_00045643_output.png b/examples/imagenet/imagenet_patches_ILSVRC2012_val_00045643_output.png