Skip to content
Generate your lover with your photo
Python Shell
Branch: master
Clone or download
Latest commit 33ae12b Jan 10, 2020
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
datasets/marriage_crop first commit Jan 6, 2020
models first commit Jan 6, 2020
preprocessing first commit Jan 6, 2020
references first commit Jan 6, 2020
utils first commit Jan 6, 2020
README.md Update README.md Jan 10, 2020
Screenshot from 2020-01-08 09-35-46.png Add files via upload Jan 7, 2020
Screenshot from 2020-01-08 09-36-04.png Add files via upload Jan 7, 2020
autotest.sh first commit Jan 6, 2020
rename.py first commit Jan 6, 2020
train.py first commit Jan 6, 2020

README.md

CoupleGenerator

Introduction

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:

image

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.

Environments

Tensorflow==1.1

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

https://github.com/machrisaa/tensorflow-vgg

Quick start

Download the training images and unzip it: https://cloudstor.aarnet.edu.au/plus/s/VWZJaWfbla3kFch

Run bash autotest.sh

Trained model (with limited data and training steps): https://cloudstor.aarnet.edu.au/plus/s/YHDWgez1g3RFc6o

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.

Extension

The marriage images were collected on BaiDu image by using a spider. Here is a very old and simple code for collecting images: https://blog.csdn.net/qq_27879381/article/details/65015280#comments

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.

AD

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.

CVPR2019: https://github.com/irfanICMLL/structure_knowledge_distillation

You can’t perform that action at this time.