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

problem training on master #584

Closed YoannRandon closed 10 months ago

YoannRandon commented 10 months ago

After I install the last version of the code, I tried to run a training using the following cmd : "python train.py --dataroot /home/shared/seg_sem_2 --checkpoints_dir /home/shared/checkpoint_model --name oktal2bdd --config_json /home/ubuntu/seg_sem.json --output_display_aim_server 127.0.0.1 --output_display_visdom_port 8501 --gpu_ids 0,1,2,3 --train_semantic_mask --data_relative_paths" I encounter this problem : image My data is structure as the following : -seg_sem_2 # name of dataset -trainA # Synthetic Oktalse -bbox # boundingbox png format -imgs # images png format -paths.txt # link imgs/bbox -trainB # realistic BDD100K -bbox # boundingbox png format -imgs # images jpg format -paths.txt

YoannRandon commented 10 months ago

furthermore, I don't understand why B domain is related to trainA when the config file I am using is AtoB direction

pierrot-lc commented 10 months ago

Hi,

This error could be due to your images being not found. From what you wrote, you should have a dataroot that follows this structure :

seg_sem_2/
├── trainA
│   ├── bbox
│   ├── imgs
│   └── paths.txt
└── trainB
    ├── bbox
    ├── imgs
    └── paths.txt

Since the error arises when you're trying to read images from domain B, it could be that your seg_seq_2/trainB/paths.txt file is wrong. It looks like it tries to search for images in the trainA/imgs directory, maybe paths are should point to the trainB/imgs directory ? (you should check if the files exists in this directory).

Hope this helps.

YoannRandon commented 10 months ago

yes i missed this. Thanks