- It's a project for building a deep learning model generating Pagesus.
- It only uses horses and birds dataset from CIFAR-10 or STL-10.
- Building a VAE model with transfer learning.
- Training the model by the dataset we processed.
\
- 2-channels encoder
- As for horses and birds dataset, we choosed the figures that birds' wings and horses' head can be seen clearly.
- "processed" is the dataset we processed by dataProcess.py
- The detailed code is in pegasus-code.ipynb
- Other details are in the report with ICLR 2021 template.