search

jingyuanli001 / RFR-Inpainting

The source code for CVPR 2020 accepted paper "Recurrent Feature Reasoning for Image Inpainting"
MIT License
355 stars 76 forks source link

Grayscale images? #52

Closed turian closed 2 years ago

turian commented 3 years ago

It appears the code only support RGB images. Can you add support for grayscale too?

kopop8 commented 3 years ago

I added img = imread(self.data[index], as_gray=False, pilmode="RGB") when readin images in load_item function in dataset.py

turian commented 3 years ago

@kopop8 that will create a 3-channel model, which will have 3x the number of parameters?

kopop8 commented 3 years ago

Thats true., this was just a quick workaround i used.

turian commented 3 years ago

@kopop8 yes thanks, I've done something like that too