Armour / Automatic Image Colorization
Licence: gpl-3.0
🎨 Automatic Image Colorization using TensorFlow based on Residual Encoder Network
Stars: ✠130
Programming Languages
Projects that are alternatives of or similar to Automatic Image Colorization
Deepway
This project is an aid to the blind. Till date there has been no technological advancement in the way the blind navigate. So I have used deep learning particularly convolutional neural networks so that they can navigate through the streets.
Stars: ✠118 (-9.23%)
Mutual labels: gpu
Libtorch Yolov5
A LibTorch inference implementation of the yolov5
Stars: ✠127 (-2.31%)
Mutual labels: gpu
Pixelfarm
From Vectors to (sub) Pixels, C# 2D Rendering Library
Stars: ✠120 (-7.69%)
Mutual labels: gpu
Finitediff.jl
Fast non-allocating calculations of gradients, Jacobians, and Hessians with sparsity support
Stars: ✠123 (-5.38%)
Mutual labels: gpu
Pex Context
Modern WebGL state wrapper for PEX: allocate GPU resources (textures, buffers), setup state pipelines and passes, and combine them into commands.
Stars: ✠117 (-10%)
Mutual labels: gpu
Yudisplacementtransition
A GPU accelerated transition library makes use of displacement maps to create distortion effects.
Stars: ✠121 (-6.92%)
Mutual labels: gpu
Colorization
Automatic line art colorization using various types of hint or without hint
Stars: ✠125 (-3.85%)
Mutual labels: colorization
Gpu Motunui
GPU-Motunui is a path tracer that renders Disney Animation's Moana Island scene.
Stars: ✠120 (-7.69%)
Mutual labels: gpu
Context
ConText v4: Neural networks for text categorization
Stars: ✠120 (-7.69%)
Mutual labels: gpu
Chromagan
Official Implementation of ChromaGAN: An Adversarial Approach for Picture Colorization
Stars: ✠117 (-10%)
Mutual labels: colorization
Ivy
The templated deep learning framework, enabling framework-agnostic functions, layers and libraries.
Stars: ✠118 (-9.23%)
Mutual labels: gpu
Ergo Pe Av
🧠🦠An artificial neural network and API to detect Windows malware, based on Ergo and LIEF.
Stars: ✠130 (+0%)
Mutual labels: gpu
Ipyexperiments
jupyter/ipython experiment containers for GPU and general RAM re-use
Stars: ✠128 (-1.54%)
Mutual labels: gpu
Automatic Image Colorization
Overview
This is a TensorFlow implementation of the Residual Encoder Network based on Automatic Colorization and the pre-trained VGG16 model from https://github.com/machrisaa/tensorflow-vgg
For latest TensorFlow with estimator support, check tf-1.12 branch. (still under development, the training code is working now)
Structure
-
config.py
: config variables like batch size, training_iters and so on -
image_helper.py
: all functions related to image manipulation -
read_input.py
: all functions related to input -
residual_encoder.py
: the residual encoder model -
common.py
: the common part for training and testing, which is mainly the workflow for this model -
train.py
: train the residual encoder model using TensorFlow built-in AdamOptimizer -
test.py
: test your own images and save the output images
TensorFlow graph
How to use
-
Download pre-trained VGG16 model vgg16.npy to vgg folder
-
Option 1: Use pre-trained residual encoder model
- Download model here
- Unzip all files to
summary_path
(you can change this path inconfig.py
)
-
Option 2: Train your own model!
- Change the
batch_size
andtraining_iters
if you want. - Change
training_dir
to your directory that has all your training jpg images - Run
python train.py
- Change the
-
Test
- Change
testing_dir
to your directory that has all your testing jpg images - Run
python test.py
- Change
Examples
-
More example output images can be found in sample_output_images folder.
References
Contributing
See CONTRIBUTING.md
License
GNU GPL 3.0 for personal or research use. COMMERCIAL USE PROHIBITED.
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].