search

anilsathyan7 / Portrait-Segmentation

Real-time portrait segmentation for mobile devices
MIT License
643 stars 133 forks source link

Train the model on another datasets #3

Closed pandamax closed 4 years ago

pandamax commented 4 years ago

thanks for sharing,if I want to train the model on another datasets,how should I prepare for the dataset?

for example,I have an original image apple.jpg that includes an apple,and another imagemask.png that includes the apple's mask.

how should I convert the dataset?thanks in advance oh,there is ! https://github.com/anilsathyan7/Portrait-Segmentation/blob/master/utils/data.py