Skip to content

haolol0606/SRCNN-Reimplementation-on-DIV2K-Datasets

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SRCNN-Reimplementation-on-DIV2K-Datasets

This repository contains a reimplementation of the SRCNN (Super-Resolution Convolutional Neural Network) for image super-resolution, trained and evaluated on DIV2K Datasets.
It is a PyTorch-based implementation that demonstrates high-quality upscaling and performance evaluation with PSNR and SSIM metrics.

Models

You can download trained SRCNN models (trained on DIV2K (x3 only)) here:

Features

  • Reimplementation of the original SRCNN architecture
  • Supports training and testing on DIV2K Datasets
  • Evaluation with PSNR and SSIM metrics
  • Save and load model checkpoints
  • Example scripts for training and inference

📊 Example Outputs

SRCNN Baseline SRCNN Improved

Left: Baseline SRCNN    |    Right: Improved SRCNN

About

Reimplement Super Resolution CNN by using DIV2K datasets

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages