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:41:18 +02:00
parent fa37768d43
commit 61b94755b6
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
README.md
View File

@@ -1,9 +1,10 @@
# Colorization of Grey Images by applying a Convolutional Autoencoder on the Jetson Nano
## by Dennis Konkol and Tim Niklas Witte
This repository contains an pretrainied convolutional autoencoder for colorization of grey images.
This repository contains an pretrainied convolutional autoencoder for colorization of grey images.
The live camera stream will be colorizatized in real time.
The architecture of the ANN is optimized to run on the Jetson Nano.
It has 300.000 parameters.
In total, 10 FPS can be archived on this embedded GPU.
![Example Video](videoPresentation.gif)
@@ -31,10 +32,6 @@ Non-trainable params: 1,092
_______________________________________________________________
```
![Example Video](./Plots/Layers/EncoderLayer.png)
![Example Video](./Plots/Layers/DecoderLayer.png)
## Usage
### Training