Human portrait drawing powered by AI (U^2-Net)

Using the recently released U²-Net trained model, it is possible to create human portrait drawing from human photo!

In this article, I will explain how to use U²-Net on Google Colaboratory. If you don’t have any knowledge of machine learning, don’t worry. I have prepared a notebook for you to experience wonderful experience of AI. To use it, you just need to follow a few steps!

  1. Visit https://colab.research.google.com/notebooks/welcome.ipynb.

2. Open notebook (File > Open notebook)

  • Select “GitHub” tab
  • Input first editbox(Enter a GitHub URL or…) to “adakoda”
  • Select “Repository” to “adakoda/U-2-Net”
  • Select notebook to “notebook/U_2_Net_demo.ipynb” and select (click line) it

3. Run all (Runtime > Run all)

  • Select “RUN ANYWAY”

The complete code for the notebook can be found in the following repository.

Enjoy!

--

--

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store