search

jolibrain / joliGEN

Generative AI Image Toolset with GANs and Diffusion for Real-World Applications
https://www.joligen.com
Other
238 stars 31 forks source link

Import images with paths.txt file in unaligned mode #589

Open JulienKszb opened 9 months ago

JulienKszb commented 9 months ago

Hello,

I have a training in unaligned mode. As I understand, the structure of the dataset directory should be like this :

domainA2domainB/
    trainA/
        image_a1.jpg
        image_a2.jpg
        ...
    trainB/
        image_b1.jpg
        image_b2.jpg
        ...

I would be interested in importing images using paths.txt files as follows:

domainA2domainB/
    trainA/
        paths.txt
    trainB/
        paths.txt

For now it is not working and it will be interesting given my infra. Is it possible to add this feature without too much effort?

beniz commented 9 months ago

Hi, you can easily implement it here: https://github.com/jolibrain/joliGEN/blob/master/data/unaligned_dataset.py from looking at https://github.com/jolibrain/joliGEN/blob/master/data/unaligned_labeled_mask_dataset.py If it works you can PR it :)

JulienKszb commented 9 months ago

Thank you, I will have look at it !