How to train on my own data?

Joker316701882 / Deep-Image-Matting

This is tensorflow implementation for paper "Deep Image Matting"

626 stars 188 forks source link

How to train on my own data? #33

Open b121351 opened 5 years ago

b121351 commented 5 years ago

I'm looking to train on a small data set I've created. I've input images, trimaps, alpha(ground truths), all in png formats. But I've seen other type of data like 'eps','bg' in matting_unpool.py. dataset_eps = 'train_data/eps' dataset_BG = 'train_data/bg'

How to proceed?