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
918 stars 200 forks source link

请问在哪里修改生成图像的名称 #134

Open TXM-MB opened 1 year ago

TXM-MB commented 1 year ago

请问在哪里修改生成图像的名称 想用NIQE进行图像质量评价 但是源文件图像和生成器保存图像的名称不一样 求指教

jiayouaa commented 10 months ago

EnlightenGAN/util/visualizer.py 中修改为 def save_images(self, webpage, visuals, image_path): image_dir = webpage.get_image_dir() short_path = ntpath.basename(image_path[0]) name = os.path.splitext(short_path)[0]

    webpage.add_header(name)
    ims = []
    txts = []
    links = []

    for label, image_numpy in visuals.items():
        if label == 'fake_B':
            # image_name = '%s_%s.png' % (name, label)
            image_name = '%s.png' % name
            save_path = os.path.join(image_dir, image_name)
            util.save_image(image_numpy, save_path)

            ims.append(image_name)
            txts.append(label)
            links.append(image_name)
    webpage.add_images(ims, txts, links, width=self.win_size)