Skip to content

luchonaveiro/color-sketch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Painting Black and White Sketches with a U-Net algorithm

Develop and train a deep learning with this Kaggle dataset. You can find a more detailed explanation on this Medium post.

Requirements:

  • tensorflow 2.0
  • numpy
  • pandas
  • matplotlib
  • tqdm
  • dash
  • plotly
  • requests
  • pillow

On the model-development folder, you can find the notebook that trains the model, it also has the Dockerfile that serves the trained model with TensorFlow Serving.

On the dash-app folder, you can find a Dash application that consumes the trained model, it also has the Dockerfile that builds the app.

You can build the entire project with the docker-compose.yml config file, by executing the following command

$ docker-compose up

Example

Here is an example of the app running:

About

U-Net for coloring black and white sketches

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages