Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Andrei Moraru committed Jul 30, 2023
1 parent 1451055 commit ae19084
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,13 @@
## Steps
1. `pip install -r requirements.txt`
2. download the [COCO dataset](https://cocodataset.org/#home) (I use COCO 2017).
3. download the [Set5, Set14, BSD100 test datasets](https://github.com/XPixelGroup/BasicSR/blob/master/docs/DatasetPreparation.md#common-image-sr-datasets)
4. [create_data_lists.py](https://github.com/AndreiMoraru123/Super-Resolution/blob/main/create_data_lists.py) creates the JSON files for training and testing from the COCO image paths.
5. [train.py](https://github.com/AndreiMoraru123/Super-Resolution/blob/main/train.py) runs the whole training pipeline with top-down logic found in the file. Everything is managed by the `Trainer` from [trainer.py](https://github.com/AndreiMoraru123/Super-Resolution/blob/main/trainer.py).
6. [resolve.py](https://github.com/AndreiMoraru123/Super-Resolution/blob/main/resolve.py) generates the super resolution images from a given high resolution image (the low resolution version to be solved is generated by down-sampling the given image) and evaluates the models using with `scikit-image`'s `peak_signal_noise_ratio` and `structural_similarity` using the `Evaluator` from [evaluator.py](https://github.com/AndreiMoraru123/Super-Resolution/blob/main/evaluator.py).

The code itself is heavily commented and you can get a feel for how super resolution models work by looking at the [tests](https://github.com/AndreiMoraru123/Neural-Machine-Translation/tree/main/test).

## Overfitting on one image

Expand Down

0 comments on commit ae19084

Please sign in to comment.