677467e2ae6a25911428540620b5e992cd64f482,src/predict/predict.py,Predictor,forward_pass,#Predictor#,54

Before Change

                imwrite(output_path, sr_img)

    def forward_pass(self, file_path):
        lr_img = imread(str(file_path)) / 255.0
        lr_img = np.expand_dims(lr_img, axis=0)
        sr_img = self.model.rdn.predict(lr_img)
        return np.clip(sr_img, 0, 1) * 255

After Change

    def forward_pass(self, file_path):
        lr_img = imread(file_path)
        if lr_img.shape[2] == 3:
            lr_img = lr_img / 255.0
            lr_img = np.expand_dims(lr_img, axis=0)
            sr_img = self.model.rdn.predict(lr_img)
            return np.clip(sr_img, 0, 1) * 255
        else:
            error(" {} is not an image with 3-dim.".format(file_path))

Instances