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-30 17:38:18 +02:00
parent eedd385635
commit fa37768d43
1 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
README.md
View File

@@ -13,6 +13,28 @@ In total, 10 FPS can be archived on this embedded GPU.
- TensorFlow 2
- OpenCV 3.3.1
## Model
```bash
Model: "autoencoder"
_______________________________________________________________
Layer (type) Output Shape Param #
===============================================================
encoder (Encoder) multiple 148155
decoder (Decoder) multiple 150145
===============================================================
Total params: 298,302
Trainable params: 297,210
Non-trainable params: 1,092
_______________________________________________________________
```
![Example Video](./Plots/Layers/EncoderLayer.png)
![Example Video](./Plots/Layers/DecoderLayer.png)
## Usage
### Training