search

sshh12 / terrain-diffusion

MIT License
17 stars 2 forks source link

How many training images are required for model training? #7

Open lntuleej opened 1 month ago

lntuleej commented 1 month ago

Thanks for your great work！i have run successful on my machine，with your code and dataset。But i try to train on my own dataset only 20 images. the inference result is failed.

sshh12 commented 1 month ago

Hey! 20 images is not enough for quality outputs -- this is especially the case for finetuning an inpainting model like this where more examples are needed for both the domain and the different kinds of masks. From https://github.com/sshh12/terrain-diffusion/blob/main/scripts/download_rgb_dataset.py I believe I had several thousand examples.

lntuleej commented 1 month ago

邮件已经收到，将尽快给您回复。