update readme

2 years ago · 1b45a5de2a
parent 985c5fad4c
commit 1b45a5de2a
1 changed files with 1 additions and 0 deletions
--- a/README.md
+++ b/README.md
@ -75,6 +75,7 @@ images = images.to('cpu').detach().numpy()
 Then image $i$ can be coverted to a PIL.Image and saved with its `save` method
 ```python
 image = Image.fromarray(images[i])
+image.save('image_{}.png'.format(i))
 ```

 ### Interactive