Esempio n. 1
0
assert model_type in ["sr", "esr", "dsr", "ddsr", "rnsr"], 'Model type must be either "sr", "esr", "dsr", ' \
                                                           '"ddsr" or "rnsr"'

mode = str(args.mode).lower()
assert mode in ['fast',
                'patch'], 'Mode of operation must be either "fast" or "patch"'

scale_factor = int(args.scale)
save = strToBool(args.save)

patch_size = int(args.patch_size)
assert patch_size > 0, "Patch size must be a positive integer"

if model_type == "sr":
    model = models.ImageSuperResolutionModel(scale_factor)
elif model_type == "esr":
    model = models.ExpantionSuperResolution(scale_factor)
elif model_type == "dsr":
    model = models.DenoisingAutoEncoderSR(scale_factor)
elif model_type == "ddsr":
    model = models.DeepDenoiseSR(scale_factor)
elif model_type == "rnsr":
    model = models.ResNetSR(scale_factor)
else:
    model = models.DeepDenoiseSR(scale_factor)

model.upscale(path,
              save_intermediate=save,
              mode=mode,
              patch_size=patch_size,
              suffix=suffix)
Esempio n. 2
0
args = parser.parse_args()

path = args.imgpath
suffix = args.suffix

model_type = str(args.model).lower()
assert model_type in ["sr", "esr", "dsr", "ddsr", "rnsr"], 'Model type must be either "sr", "esr", "dsr", ' \
                                                           '"ddsr" or "rnsr"'

mode = str(args.mode).lower()
assert mode in ['fast', 'patch'], 'Mode of operation must be either "fast" or "patch"'

scale_factor = int(args.scale)
save = strToBool(args.save)

patch_size = int(args.patch_size)
assert patch_size > 0, "Patch size must be a positive integer"

if model_type == "sr":
    model = models.ImageSuperResolutionModel()
elif model_type == "esr":
    model = models.ExpantionSuperResolution()
elif model_type == "dsr":
    model = models.DenoisingAutoEncoderSR()
elif model_type == "ddsr":
    model = models.DeepDenoiseSR()
elif model_type == "rnsr":
    model = models.ResNetSR()

model.upscale(path, scale_factor=scale_factor, save_intermediate=save, evaluate=False, mode=mode,
              patch_size=patch_size, suffix=suffix)