This is a benchmark project to make higher resolution versions of low resolution flower images:
- Be sure to sign up for W&B.
- Clone this repository:
git clone https://github.com/wandb/superres.git
- Run
pip install -U -r requirements.txt
to install requirements. - Run
python train.py
to train the baseline model. Modify this file and the data pipeline (or write your own scripts and create different model architectures!) to get better results. - Submit your results to the benchmark.
The dataset is comprised of images of flowers. The training set has 5000 images of flowers that have been resized. The test set has 670 images of flowers. The input size is 32x32 pixels and the output size is 256x256 pixels.
The goal is to enhance a low resolution input image to be 8 times greater resolution with the least loss of quality.
We use a perceptual distance metric (val_perceptual_distance) on the validation set to rank results (lower values are better).
You can submit your best runs to our benchmark. More specifically, go the "Runs" table in the "Project workspace" tab of your project. Hover over the run's name, click on the three-dot menu icon that appears to the left of the name, and select "Submit to benchmark".
- Implement a GAN
- Different loss functions
- Data augmentation