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

training with new dataset #15

Closed
javismiles opened this issue Oct 2, 2020 · 11 comments
Closed

training with new dataset #15

javismiles opened this issue Oct 2, 2020 · 11 comments

Comments

@javismiles
Copy link

Good day
I want to try to train the model with a new dataset I´m creating,

In this new dataset, for each input image, I have a ground truth alpha mask.

In here:
train_image_file = ImageFileTrain(alpha_dir=CONFIG.data.train_alpha,
fg_dir=CONFIG.data.train_fg,
bg_dir=CONFIG.data.train_bg)

It seems to ask for 3 folders, one for the ground truth alpha it seems, and then two for fg and bg?
what are these?

if I have input images and ground truth alphas, how can I adapt them to work with the input of your model?
thank you very much

@Yaoyi-Li
Copy link
Owner

Hi, fg means 'foreground images' and bg is 'background'. Your input images are foreground images and you need one more folder for background images. You can refer to our paper or Deep Image Matting for more details about the composition.

@javismiles
Copy link
Author

javismiles commented Oct 14, 2020 via email

@Yaoyi-Li
Copy link
Owner

Yes, almost the same as you said, and we composite images during the training. The number of samples in each epoch is the same as background images.

@javismiles
Copy link
Author

javismiles commented Oct 19, 2020 via email

@Yaoyi-Li
Copy link
Owner

It highly depends on your CPUs and the complexity of your augmentations. In our training, we are doing the compositing after the resize operation, which is faster than merging high-resolution images off-line. It is a little slower but totally acceptable.

@javismiles
Copy link
Author

javismiles commented Oct 19, 2020 via email

@javismiles
Copy link
Author

javismiles commented Oct 19, 2020 via email

@javismiles
Copy link
Author

javismiles commented Oct 20, 2020 via email

@javismiles
Copy link
Author

javismiles commented Oct 28, 2020 via email

@javismiles
Copy link
Author

javismiles commented Oct 28, 2020 via email

@javismiles
Copy link
Author

javismiles commented Oct 28, 2020 via email

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

No branches or pull requests

2 participants