expand readme

This commit is contained in:
TimNiklasWitte
2022-03-31 10:09:33 +02:00
parent 67a43ad600
commit 8793724bc0
2 changed files with 16 additions and 2 deletions

View File

@@ -0,0 +1,10 @@
# Colorful Image Colorization
## by Richard Zhang, Phillip Isola, Alexei A. Efros
Link to the paper: https://arxiv.org/abs/1603.08511
For evaluation purpose, the entire code in this directory was copied from this repo:
https://github.com/foamliu/Colorful-Image-Colorization
We compared the results generated by the model form Richard Zhang et al. with the results of our model.

View File

@@ -9,6 +9,10 @@ In total, 10 FPS can be archived on this embedded GPU.
![Example Video](videoPresentation.gif)
In entire documentation of this project in form of a paper can be find in
`./Paper/Main.pdf`.
This project was created as a part of the GPU programming course of Mario Porrmann in the winter term 2021/22 of the Osnabrück University.
## Requirements
- TensorFlow 2
@@ -35,7 +39,7 @@ _______________________________________________________________
### Hyperparameters
```bash
```python3
optimizer = Adam
learning rate = 0.0001
loss function = mean squared error
@@ -49,7 +53,7 @@ batch size = 32
Run `Training.py` to start the training of the model on the `imagenet2012` dataset.
Each epoch the weights are stored into `./saved_models`.
Besides, in `./test_logs` are the corresponding trainings statistics (train and test loss and also a batch of colorized test images) logged.
Note that, the `imagenet2012` dataset is stored in `./imagenet` as described in this [blog article](https://towardsdatascience.com/preparing-the-imagenet-dataset-with-tensorflow-c681916014ee)
Note that, the `imagenet2012` dataset is stored in `./imagenet` as described in this [blog article](https://towardsdatascience.com/preparing-the-imagenet-dataset-with-tensorflow-c681916014ee).
```bash
python3 Training.py