Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

One question on result of train #6

Closed
hsysfan opened this issue Jul 24, 2023 · 2 comments
Closed

One question on result of train #6

hsysfan opened this issue Jul 24, 2023 · 2 comments
Labels
good first issue Good for newcomers

Comments

@hsysfan
Copy link

hsysfan commented Jul 24, 2023

Hello, im a researcher on data augmentation for defect images

I trained your model(DFMGAN) twice for two kind of custom data

First was showing very good result.

But First stage(normal image training) of second data shows some curious results.

The trained model generate rotated image from normal dataset images.

image

Then, i found the augmentation options on training_options.json

There is "rotate90 : 1" on the file.

Can you explain why DFMGAN generates rotated images?

@Ldhlwh
Copy link
Owner

Ldhlwh commented Jul 24, 2023

Hello @hsysfan,

This issue is probably due to the usage of adaptive augmentation in StyleGAN2-ADA. Such rotation is prone to happen when the defect-free dataset is in small quantity or low diversity since the probability of the augmentation will be high trying to counterweight the overfitting.

According to our previous experiments, such rotation will not be happening all the time. You may choose the snapshot at another kimg, try to rerun the training process, or simply turn off the adaptive augmentation if this issue keeps showing up.

@CVKim
Copy link

CVKim commented Aug 21, 2024

@hsysfan

image

I'm currently in the middle of training, and I noticed that some images are being rotated, which seems to be disrupting the training. I disabled the rotate90 and rotate sections in the code, but the training still isn't going well. The results for the normal images are fine, but the rotated images are causing issues.

Has this issue been resolved?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
good first issue Good for newcomers
Projects
None yet
Development

No branches or pull requests

3 participants