vijishmadhavan / SkinDeep

Get Deinked!!

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

waving

SkinDeep   Tweet  

Contact: vijishmadhavan@gmail.com

You can sponsor me to support my open source work 💖 sponsor

Updates

SkinDeep Video is getting ready, thanks to 3dsf (https://github.com/3dsf) for the tremendous effort he has put into this project to make this happen.

post

I planned this project after watching Justin Bieber's "Anyone" Music Video, He had his tattoo covered up with the help of artists airbrushing on him for hours. The results were amazing in the music video. Producing that sought of video output can be difficult, so I opted for Images. Can deep learning do a decent job or can it even match photoshop? This was the starting point of this project!!

Why not photoshop?

Photoshop can produce extremely good results but it needs expertise and hours of work retouching the whole image.

Highlights

Synthetic data generation

To do such a project we need a lot of image pairs, I couldn't find any such dataset so I opted for synthetic data.

(1) Overlaying APDrawing dataset image pairs along with some background removed tattoo designs, This can be easily done using Python OpenCV.

(2) Apdrawing dataset has line art pairs which will mimic tattoo lines, this will help the model to learn and remove those lines.

(3) APDrawing dataset only has portrait head shots, For full body images I ran my previous ArtLine(https://github.com/vijishmadhavan/ArtLine) project and overlaid the output with the input image.

Imgur

Imgur

(4) ImageDraw.Draw was used with forest green colour codes and placed randomly on zoomed-in body images, Similar to Crappify in fast.ai.

(5) Photoshop was also used to place tattoos in subjects where warping and angle change was needed.

Imgur

Mail me for a modified Apdrawing dataset.

Example Outputs

Imgur

Imgur

Imgur

Heavily tattooed faces

Imgur

Imgur

Visual Comparison

Imgur

Technical Details

The highlight of the project is in producing synthetic data, thanks to pyimagesearch.com for wonderful blogs. Check below links.

https://www.pyimagesearch.com/2016/03/07/transparent-overlays-with-opencv/

https://www.pyimagesearch.com/2016/04/25/watermarking-images-with-opencv-and-python/

https://www.pyimagesearch.com/2015/01/26/multi-scale-template-matching-using-python-opencv/

Required libraries

This project is built around the wonderful Fast.AI library.

  • fastai==1.0.61 (and its dependencies). Please don't install the higher versions
  • PyTorch 1.6.0 Please don't install the higher versions

Getting Started Yourself

The project is still a work in progress, but I want to put it out so that I get some good suggestions.

The easiest way to get started is to simply try out on Colab:

The output is limited to 500px and it needs high quality images to do well.

I would request you to have a look at the limitations given below.

Docker

Note: These instructions are for running with an NVDIA GPU on Linux.

Prequisites

  • Make sure your NVIDA drivers are correctly installed. Running nvidia-smi should confirm your driver version and memory usage
  • Approximately 3.7GB of free GPU memory
  • The nvidia-docker2 package installed. This will allow the Docker container access to the GPU
  • Docker installed with the correct permissions for your system

Run the container

This runs against all of the GPUs. Check the Docker documentation for running on a specific card if you have multiple.
The type=bind mounts the current directory (where you checked out the project) as the home in the container. This allows you immediate access to the notebooks.

$> docker run --gpus all \
              --mount type=bind,source=`pwd`,target=/home/jovyan \
              -d \
              -p 8888:8888 -p 4040:4040 -p 4041:4041 \
              jupyter/tensorflow-notebook:python-3.8.8

Launch the site

Use 'ps' to get the id of your new container, then print out the logs to get the launch URL. This is required because an authenication token is created at startup.

$> docker ps
$> docker logs <container_id>

Limitations

  • Synthetic data does not match real tattoos, so the model struggles a bit with some images.
  • Building a huge dataset by myself was impossible, so I had to go with a limited number of image pairs.
  • Tattoo designs are unique, it differs from person to person so the model might fall in some cases.
  • Coloured tattoos won't work, the dataset had none.

Going Forward

The model still struggles and needs a lot of improvement, if you are interested please contribute lets improve it.

Acknowledgments

About

Get Deinked!!

License:Apache License 2.0


Languages

Language:Jupyter Notebook 96.9%Language:Python 3.1%Language:Dockerfile 0.0%Language:Shell 0.0%Language:Procfile 0.0%