This project is a Python-based implementation of the Super-Resolution Generative Adversarial Network (SRGAN) using PyTorch. SRGAN is a pivotal development in the field of computer vision, enabling the enhancement of image resolution beyond traditional methods. By leveraging adversarial networks, SRGAN attempts to reconstruct low-resolution images into high-definition outputs. This repository not only implements the core technology but also provides tools for training and testing the model with custom datasets.
This section lists the major frameworks and libraries that you need to run the project:
- Python - The programming language used.
- PyTorch - The deep learning framework.
- OpenCV - For image manipulation operations.
- NumPy - For high-performance scientific computing.
Ensure you have the following installed on your system:
- Python 3.8 or newer
- pip (Python package installer)
sudo apt-get install python3-pip
To set up the project locally, follow these steps:
-
Clone the repository
git clone https://github.com/iamFury2K/SR-GAN_Scratch.git cd SR-GAN_Scratch
-
Install required Python libraries
pip install -r requirements.txt
Here's how you can use this project to upscale images:
-
Prepare your dataset
Place your low-resolution images in a folder named
data/input
. -
Train the model
To train the model on your dataset, run:
python train.py
Adjust the parameters as necessary.
We welcome contributions from the community. Here are some ways you can contribute:
- Reporting bugs
- Suggesting enhancements
- Writing code for fixed bugs or added features
- Improving documentation
To contribute, please follow these guidelines:
- Fork the project repository.
- Create a new branch for your feature or fix (
git checkout -b feature/YourFeature
orgit checkout -b fix/YourBugFix
). - Commit your changes with a descriptive message.
- Push the branch to your fork.
- Open a pull request to our project.
This project is licensed under the MIT License - see the LICENSE.txt file for details.
- Aladdin Persson for foundational resources and tutorials.
- Your Tutorial Link - For providing the tutorial that inspired this project.
- Thanks to all contributors who have participated in this project.
This README is structured to provide all necessary information to get started, understand, and contribute to the project effectively.