13287043443135f6f3f8ed787d3978866c390d81,test.py,,,#,71

Before Change

            input = data["input"]
            if not args.cpu:
                input = input.cuda()
            output = model(input, args.scale).cpu() + data["bicubic"]
            sr_img = tensor2im(output, mean, stddev)
            toc = time.time()
            if "target" in data:

After Change

            tic = time.time()
            // split image in chuncks of max-dimension
            if args.max_dimension:
                data_chunks = DataChunks({"input":data["input"]},
                                         args.max_dimension,
                                         args.padding,args.scale)
                for chunk in data_chunks.iter():
                    input = chunk["input"]
                    if not args.cpu:
                        input = input.cuda()
                    output = model.predict(input,{},args.scale)
                    data_chunks.gather(output)
                output = data_chunks.concatenate() + data["bicubic"]
            else:
                input = data["input"]
                if not args.cpu:
                    input = input.cuda()
                output = model.predict(input,data,args.scale)
            sr_img = tensor2im(output, mean, stddev)
            toc = time.time()
            if "target" in data:
                hr_img = tensor2im(data["target"], mean, stddev)

Instances