Skip to content
This repository has been archived by the owner on Jun 16, 2023. It is now read-only.

Re-Implementation of SRGAN with symmetric padding for a better merged final image (Anysized input with 4x upscaled)

Notifications You must be signed in to change notification settings

vTuanpham/SRGAN_Reimplementation_modified

Repository files navigation

Keras SRGAN Reimplementation with modified Generator

Keras Re-Implementation of SRGAN with symmetric padding for a better merged final image (Anysize input with 4x upscaled)

AI-upscale application with user-interface for anysize input image with 4x upscaled output based on SRGAN with modified-Generator. Added Symmetric padding for a seamless final merged image, add more conv layers as well as more residual blocks, lay out general workflow for anysize input.

Modified-Generator image

Original Discriminator image

Pipeline image

NOTE: Only support for size up to max dim of 512 (Since perdiction is using predict_on_batch with max tensor size of (16,64,64,3) (that why we are using CPU for prediction to avoid OORam)) if you are using a more powerful CPU or planning to reimplement the code with (1,16,16,3) size input tensor per splited sub image and vstack them later, you can defined the maximum size input by changing the func computeApporiateSize

image