Catfusion conststs of 2 main parts. the noise scheduuler for adding noise to the image and a U-Net for doing the Backward process
the noise scheduler works by adding a random noise to the image based on the time step
the U-Net consists of 5 down and 5 up layers with 10 dense layers for time embeddings that are added in each layer to the output
I trained the model on my personal GPU for 2 epochs
The dataset id made up of 7000 Images collected fom Cat API Because the quality of Images wasn't good enough, I manually selected them
The results weren't very satisfying. this is as close as I could get to an Image of a Cat
Catfusion is published under CPL (Cat Public License). You are allowed to copy, modify and redistribute this software as long as the software you're using it in is related to cats and is available to the public under the same license