Python batch_numpy_to_image 예제들

예제 #1

파일: sparnet_model.py 프로젝트: luztxp/Face-SPARNet

    def get_current_visuals(self, size=128):
        out = []
        out.append(utils.tensor_to_numpy(self.img_LR))
        out.append(utils.tensor_to_numpy(self.img_SR))
        out.append(utils.tensor_to_numpy(self.img_HR))
        visual_imgs = [utils.batch_numpy_to_image(x, size) for x in out]

        return visual_imgs

예제 #2

    def get_current_visuals(self, size=512):
        out = []
        visual_imgs = []
        out.append(utils.tensor_to_numpy(self.img_LR))
        out.append(utils.tensor_to_numpy(self.img_SR))
        out.append(utils.tensor_to_numpy(self.img_HR))

        out_imgs = [utils.batch_numpy_to_image(x, size) for x in out]

        visual_imgs += out_imgs
        visual_imgs.append(utils.color_parse_map(self.ref_mask_onehot, size))
        visual_imgs.append(utils.color_parse_map(self.hr_mask, size))

        return visual_imgs

예제 #3

파일: parse_model.py 프로젝트: sailor002/PSFRGAN

    def get_current_visuals(self, size=512):
        out = []
        visual_imgs = []
        out.append(utils.tensor_to_numpy(self.img_LR))
        out.append(utils.tensor_to_numpy(self.img_SR))
        out.append(utils.tensor_to_numpy(self.img_HR))
        out_imgs = [utils.batch_numpy_to_image(x, size) for x in out]

        visual_imgs.append(out_imgs[0])
        visual_imgs.append(out_imgs[1])
        visual_imgs.append(utils.color_parse_map(self.pred_Parse))
        visual_imgs.append(utils.color_parse_map(self.gt_Parse.unsqueeze(1)))
        visual_imgs.append(out_imgs[2])

        return visual_imgs