/FIAI

Primary LanguageJupyter Notebook

Implementation of ESRGAN using RRDB x4 architecture and VGG19

create a conda virtual environment using the command
conda create -n esrgan python3.11.3

Add the packages and dependencies using pip install -r requirements.txt

Create a new directory in the project folder and name it model

Download the pretrained model weights from the Google Drive Link and paste them in the model folder

To run the pretrained model enter the command python test_remote_finetune.py

To train the model without pretrained weights enter the command python train_remote.py

To train the model with pretrained weights enter the command python finetune_remote.py