search

lyndonzheng / Pluralistic-Inpainting

[CVPR 2019]: Pluralistic Image Completion
676 stars 145 forks source link

Using multiple external (custom) irregular masks #34

Closed qwerdbeta closed 4 years ago

qwerdbeta commented 4 years ago

Hello, I am having trouble using my 5 custom masks. I set:

parser.add_argument('--mask_type', type=int, default=[3])

And I know the color requirements for masks via https://github.com/lyndonzheng/Pluralistic-Inpainting/issues/2

How do I use the following?

parser.add_argument('--mask_file', type=str, default='none', help='load test mask')

It's a string. How do I point it to multiple mask files? Thanks!

qwerdbeta commented 4 years ago

It seems as though I can point to the directory and use those masks with an flist file pointing to them like other directories of file in this project.

However, when I look at the training images, the masks appear slanted. How do I turn this rotation off?

thanks

qwerdbeta commented 4 years ago

figured it out. https://github.com/lyndonzheng/Pluralistic-Inpainting/blob/master/dataloader/data_loader.py lines 66-71 dfine transform and I just commented out the flip and rotate. It seems like the arguments are not wired up to this code.