update readme
This commit is contained in:
parent
870cebdaaf
commit
78c20ce731
10
README.md
vendored
10
README.md
vendored
|
@ -68,7 +68,15 @@ images = model.generate_images(
|
||||||
)
|
)
|
||||||
```
|
```
|
||||||
|
|
||||||
Note: you will have to move the images to the cpu and convert to numpy, e.g. `images = images.to('cpu').detach().numpy()`. Then image $i$ can be coverted to a PIL.Image `image = Image.fromarray(images[i])`, and saved with its `save` method `image.save('image.png')`.
|
Note: you will have to move the images to the cpu and convert to numpy, e.g.
|
||||||
|
```python
|
||||||
|
images = images.to('cpu').detach().numpy()
|
||||||
|
```
|
||||||
|
Then image $i$ can be coverted to a PIL.Image
|
||||||
|
```python
|
||||||
|
image = Image.fromarray(images[i])
|
||||||
|
```
|
||||||
|
and saved with its `save` method `image.save('image.png')`.
|
||||||
|
|
||||||
### Interactive
|
### Interactive
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user