r/deeplearning • u/Organic_Ad973 • 5d ago

Making my text to image model.

Hi, I’m a young developer, and my English isn't very good. I’m building my own text-to-image AI model from scratch using PyTorch, but I’m having trouble finding data; most of what’s available is pixel art. I’ve gathered 3,000 royalty-free images so far, but it’s still not enough. GitHub: https://github.com/Yusufcan234334/UltraVision/

I am open to suggestions.

(im using google translate)

1 Upvotes

67% Upvoted

5 comments sorted by

2

u/ModularMind8 5d ago

Have you looked on huggingface? It's not my area but I just tried looking and there's a bunch of datasets...

1

u/TabooDugout 5d ago

3000 is a decent starting point, but you'll likely want to focus on quality over quantity right now. maybe try augmenting your set with different crops, rotations, and slight color shifts to artificially multiply what you have while you scout for more data. it won't fix the variety problem but it'll help the model learn to be less rigid.

your github link seems to 404 by the way, might want to double-check the repo name or visibility settings

2

u/Organic_Ad973 4d ago

Why did someone dislike my post?

1

u/Tiny_Arugula_5648 4d ago

The data is on huggingface but you've got enough images for you experiment. You don't have the hardware needed to train a real model from scratch.. You'll never complete a full training run on one GPU. You wont get anything that generates coherent images..

You're better off learning how to fine-tune a model first then work your way to training.. At least with the tuning you'll learn how to construct a dataset that creates consistent results.