mirror of
https://github.com/denshooter/gpu_colorization.git
synced 2026-01-21 12:32:57 +01:00
expand readme
This commit is contained in:
10
Colorful_Image_Colorization/README.md
Normal file
10
Colorful_Image_Colorization/README.md
Normal 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.
|
||||
@@ -9,6 +9,10 @@ In total, 10 FPS can be archived on this embedded GPU.
|
||||
|
||||

|
||||
|
||||
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
|
||||
|
||||
Reference in New Issue
Block a user