Git Product home page Git Product logo

dreampower's Introduction

DreamPower

DreamPower

Available for Windows, Linux and Mac.

PRs Welcome

Download

GitHub All Releases

About

DreamPower is a fork of the DeepNude algorithm that generates better fake nudes and puts at your disposal a command line interface.

It consists of several algorithms that together create a fake nude from a photo.

If you don't have experience using command line applications you can download DreamTime which offers you a friendly user interface.

Features

DreamPower DeepNude
Multiplatform ✔️
Command-line interface ✔️
NVIDIA GPU Support ✔️
Multithreading ✔️
Automatic Scale ✔️
GIF Support ✔️
Video Support ✔️
Body Customization ✔️
Daemon ✔️
Custom Masks ✔️
Active Development ✔️

Requirements

  • 64 bits operating system:
    • Windows 7 or superior.
    • Ubuntu 16.04+
    • macOS Catalina or superior.
  • 12 GB of RAM.

GPU (Optional)

Usage

In the command line terminal run:

dreampower run --help

This will print out help on the parameters the algorithm accepts.

The input image should be 512px * 512px in size (parameters are provided to auto resize/scale your input).

Community

Supporting

DreamPower is an open-source project that will be free forever. The project is kept in development thanks to the support of our incredible backers, you can also help keep the project alive in different ways:


How does DreamPower work?

DreamPower uses an interesting method to solve a typical AI problem, so it could be useful for researchers and developers working in other fields such as fashion, cinema and visual effects.

The algorithm uses a slightly modified version of the pix2pixHD GAN architecture. If you are interested in the details of the network you can study this amazing project provided by NVIDIA.

A GAN network can be trained using both paired and unpaired dataset. Paired datasets get better results and are the only choice if you want to get photorealistic results, but there are cases in which these datasets do not exist and they are impossible to create. A database in which a person appears both naked and dressed, in the same position, is extremely difficult to achieve, if not impossible.

We overcome the problem using a divide-et-impera approach. Instead of relying on a single network, we divided the problem into 3 simpler sub-problems:

    1. Generation of a mask that selects clothes
    1. Generation of a abstract representation of anatomical attributes
    1. Generation of the fake nude photo

Original problem:

Dress To Nude

Divide-et-impera problem:

Dress To Mask Mask To MaskDet MaskDeto To Nude

This approach makes the construction of the sub-datasets accessible and feasible. Web scrapers can download thousands of images from the web, dressed and nude, and through photoshop you can apply the appropriate masks and details to build the dataset that solve a particular sub problem. Working on stylized and abstract graphic fields the construction of these datasets becomes a mere problem of hours working on photoshop to mask photos and apply geometric elements. Although it is possible to use some automations, the creation of these datasets still require great and repetitive manual effort.

Computer Vision Optimization

To optimize the result, simple computer vision transformations are performed before each GAN phase, using OpenCV. The nature and meaning of these transformations are not very important, and have been discovered after numerous trial and error attempts.

Considering these additional transformations, the phases of the algorithm are the following:

  • dress -> correct [OPENCV]
  • correct -> mask [GAN]
  • mask -> maskref [OPENCV]
  • maskref -> maskdet [GAN]
  • maskdet -> maskfin [OPENCV]
  • maskfin -> nude [GAN]

Transformations

dreampower's People

Contributors

1337roy avatar deeppppp avatar dependabot[bot] avatar kolessios avatar pommedroid avatar ringo-nine avatar stacklikemind avatar wisp101 avatar www439198341 avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.