Skip to content
New issue

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

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

Already on GitHub? # to your account

Bias Image Inpainting to paint water #77

Closed
gcosne opened this issue Jun 18, 2019 · 7 comments
Closed

Bias Image Inpainting to paint water #77

gcosne opened this issue Jun 18, 2019 · 7 comments

Comments

@gcosne
Copy link
Member

gcosne commented Jun 18, 2019

Exploring inpainting methods such as : Generative Image Inpainting to paint water.

@gcosne
Copy link
Member Author

gcosne commented Jun 26, 2019

To do :

  1. Images must be constant size
  2. Adapt code to use custom masks

It seems that masks are randomly generated and some modifications of the data loader are required to use custom masks : 275.

Author suggests having a look at neuralgym.data_from_fnames. cf neuralgym

@gcosne
Copy link
Member Author

gcosne commented Jun 27, 2019

Actual version of the code I'm modifying is the V2. The first version of the paper & code was designed for mask of rectangular shape while v2 works well with any mask.
image

Current state: - Training with random mask and monitoring tensorboard on slurm is working.

But I have several issues while loading custom mask with the neuralgym dataloader
error : 'data_from_fnames.py:153: image is None' .

Suggestion:

  • Building a custom dataloader for image-mask pairs.
  • The difficulty lies in not breaking the data-augmentation procedure.

@sashavor
Copy link
Contributor

sashavor commented Jun 27, 2019 via email

@sashavor
Copy link
Contributor

sashavor commented Jun 27, 2019 via email

@gcosne
Copy link
Member Author

gcosne commented Jun 27, 2019

I shamefully had an issue with the path & number of channels, I still have some issues with data augmentation here but it works if I disable it.

First epochs..
image

image

@sashavor
Copy link
Contributor

sashavor commented Jun 28, 2019 via email

@gcosne
Copy link
Member Author

gcosne commented Jul 22, 2019

Link to the code: https://github.com/cc-ai/generative_inpainting

GitHub
Generative Image Inpainting with Contextual Attention https://arxiv.org/abs/1801.07892, demo http://jiahuiyu.com/deepfill - cc-ai/generative_inpainting

# for free to join this conversation on GitHub. Already have an account? # to comment
Projects
None yet
Development

No branches or pull requests

3 participants