Skip to content

bijmuj/SketchColorization

Repository files navigation

SketchColorization

Trying to do AI based sketch line art colorization. Mostly focused on style transfer.

For info on all the runs and logs: W&B project link

Neural Style Transfer

I don't remember why I didn't try this initially. Not that it matters, the results are still not up to par. The outlines have become much fuzzier and it fails to keep colors within them. Credit for the code goes to Tensorflow Tutorials.

NST example

AE

Tried using a simple autoencoder as a baseline. Did not work out well. Images are blurry and might as well be grayscale.

AE example

V1

Based off of this paper. Used a UNet Generator, DCGAN-like Discriminator and VGG19 for style extraction.

Has trouble keeping within the outlines and taking all the information from the style hint.

v1_example

V2

Extremely experimental (mostly throwing things at a wall and seeing what sticks). Similar architecture to v1 but taking style hint from the last few MaxPool layers of VGG19 rather than fc1.

Also changed the way we get color style, instead of using the color image of the sketch as the style, I used some other color image in the batch.

Blurry, with seemingly nothing being done with the style information.

v2_example

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published