search

VITA-Group / EnlightenGAN

[IEEE TIP] "EnlightenGAN: Deep Light Enhancement without Paired Supervision" by Yifan Jiang, Xinyu Gong, Ding Liu, Yu Cheng, Chen Fang, Xiaohui Shen, Jianchao Yang, Pan Zhou, Zhangyang Wang
Other
890 stars 198 forks source link

Running inference on a single image #32

Closed nelaturuharsha closed 4 years ago

nelaturuharsha commented 4 years ago

Hello -- So my use case is this : Take in a single input image, run it through EngligtenGAN and get a single input image as output. A few questions:

  1. Is 200_net_G_A.pth only for aligned/unaligned datasets with an input_A and an input_B?
  2. In case I'd like to work with a single image, could I work with these weights, or would I need to train my model from scratch?
  3. I'm assuming the test model generator for a single image would be unet_256, or is there a different generator I should use.

I'd be grateful for the help.

yifanjiang19 commented 4 years ago

To run it on single images, just put your data on ../test_dataset/testA, and put another whatever image on ../test_dataset/testB. And run command provided on README.

Default model is enough.

nelaturuharsha commented 4 years ago

Got it, thank you so much! Was able to solve my problem following what you mentioned^