denshooter/gpu_colorization
1
0
Fork 0
mirror of https://github.com/denshooter/gpu_colorization.git synced 2026-01-21 12:32:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

expand readme

Browse Source
This commit is contained in:
TimNiklasWitte
2022-03-31 14:29:26 +02:00
parent 5f2b04e834
commit 6671c470a4
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@@ -92,6 +92,9 @@ python3 live_recolor_plot.py
### Pretrained Model
The model was runned for 13 epochs on the `imagnet2012` dataset and its weights are stored in `./saved_models`.
![Train and test loss](./Plots/TrainTestLossPlot.png)
Note that, grey images must have a shape of `(256,256,1)`.
The following code will load the pretrained model and colorized an image:
@@ -105,5 +108,5 @@ grey_img = ... # grey_img.shape = (256,256,1)
grey_img = np.expand_dims(grey_img, axis=0) # add batch dim
ab_img = autoencoder(grey_img) # get ab values
rbg_img = getRGB(grey_img, ab_img) # contained in Main.py
rbg_img = getRGB(grey_img, ab_img) # contained in Training.py
```