search

anilsathyan7 / Portrait-Segmentation

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

SlimNet #20

Closed pantingchn closed 3 years ago

pantingchn commented 3 years ago

Hi, i just wonder where can l get the training codes for SlimNet as j want to train the SlimNet by myself.

anilsathyan7 commented 3 years ago

Please see the slim512.ipynb notebook in the repo for training and conversion

anilsathyan7 commented 3 years ago

To preprare the dataset for training refer : https://github.com/anilsathyan7/Portrait-Segmentation/issues/22#issuecomment-688316817