search

plemeri / transparent-background

This is a background removing tool powered by InSPyReNet (ACCV 2022)
MIT License
791 stars 87 forks source link

Fine tuning / train script? #48

Closed eafpres closed 10 months ago

eafpres commented 10 months ago

Hi--I reviewed the code base, and it seems there is a pre-trained checkpoint, but I don't see any training script. I'm wondering if there could be an option to fine-tune on my own data.

Thanks again for this great work.

plemeri commented 10 months ago

You can train your model from our original repository InSPyReNet. After training, you can plug-in your model. Please refer to https://github.com/plemeri/transparent-background?tab=readme-ov-file#new-configuration.

Thanks.

plemeri commented 10 months ago

Closing due to inactivity. Please open another issue if problem persists.