logo
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions

80 lines
1.7 KiB

3 years ago
# Animating using AnimeGanV2
*author: Filip Haltmayer*
3 years ago
3 years ago
## Description
Convert an image into an animated image using [`AnimeganV2`](https://github.com/TachibanaYoshino/AnimeGANv2).
## Code Example
Load an image from path './test.png'.
3 years ago
*Write the pipeline in simplified style*:
```python
import towhee
towhee.glob('/Users/chenshiyu/workspace/data/pic/test.png') \
.image_decode.cv2() \
.img2img_translation.animegan(model_name = 'hayao') \
.show()
3 years ago
```
<img src="./results1.png" alt="results1" style="height:40px;"/>
*Write a same pipeline with explicit inputs/outputs name specifications:*
```python
import towhee
towhee.glob['path']('/Users/chenshiyu/workspace/data/pic/test.png') \
.image_decode.cv2['path', 'origin']() \
.img2img_translation.animegan['origin', 'transformed'](model_name = 'hayao') \
.select('origin', 'transformed') \
.show()
```
<img src="./results2.png" alt="results1" style="height:50px;"/>
3 years ago
## Factory Constructor
Create the operator via the following factory method
***img2img_translation.animegan(model_name = 'which anime model to use')***
Model options:
- celeba
- facepaintv1
- facepaintv2
- hayao
- paprika
- shinkai
## Interface
Takes in a numpy rgb image in channels first. It transforms input into animated image in numpy form.
**Parameters:**
***model_name***: *str*
​ Which model to use for transfer.
***framework***: *str*
​ Which ML framework being used, for now only supports PyTorch.
**Returns**: *numpy.ndarray*
​ The new image.
## Reference
Jie Chen, Gang Liu, Xin Chen
"AnimeGAN: A Novel Lightweight GAN for Photo Animation."
ISICA 2019: Artificial Intelligence Algorithms and Applications pp 242-256, 2019.