kskin / WaterGAN

Source code for "WaterGAN: Unsupervised Generative Network to Enable Real-time Color Correction of Monocular Underwater Images"

Home Page:https://arxiv.org/abs/1702.07392

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Model Test and code mistyping

ZhihuaLiuEd opened this issue · comments

Hi Katie:

Thanks for sharing your code.

  1. If I just change the "is_train" to False to proceed model testing. Feels like it still uses the training data to do the test? As the model want to load checkpoint with "self.model_dir", but the "self.water_dataset_name" is coming from the param in my python command line. Therefore in order to get the saved checkpoint, I have to input the training dataset during testing.

  2. modelmhl.py line 506, "out_name3 = self.resuts_dir + out_file3" -> "out_name3 = self.results_dir + out_file3"

Cheers

Zhihua