PETGreen / CoupleGenerator

Generate your lover with your photo

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool



Generate your lover with your photo

This project is an old project from two years ago. I accidentally found it in my storage medium.

We collect pictures of married couples on the Internet and pre-process the images.

The model was trained by using the code of pix2pix.

An example of training images:


You can make photos of yourself with the person you loved as a training pair~

This repo is just for entertainment.

If you have any interests (not for business), please feel free to use the code and the dataset.

Fitting the training data

Here are the results of the training set after 8800 steps. image

The model can fit the training images in a short time.



You also need to download the pretrain weight for vgg through following links:

Quick start

Download the training images and unzip it:

Run bash

Trained model (with limited data and training steps):

We use all the data for training and testing, it is not scientific for research. The motivation of this project is a surprise for my friend. We want to overfit her with the boy she loves secretly. Therefore we do not consider the generalization of the model.


The marriage images were collected on BaiDu image by using a spider. Here is a very old and simple code for collecting images:

Just change the keyword.

Thanks GZHermit for helping me to process the data. He detected the face from the collecting images and re-located the couple into training pairs.

The images we provide is just for fun, not for business activities

If you really want to train an effective model for business, you have to collect more images in legal way.


This project is a toy project, and I am working on semantic segmentation, instance segmentation on 2D images and videos for my Ph.D. career. If you are interested in these topics, you can follow my github. I will release a new code on video segmentation recently.


ezoic increase your site revenue


Generate your lover with your photo


Language:Python 99.9%Language:Shell 0.1%