train custom dataset

[plemeri]

train custom dataset

[plemeri]

Hi @crapthings, I saw your issue on our python API transparent-background. Thank you for selecting our work!

Here are the answers for your questions.

1. You can use --resume argument for training script. Put the pre-trained checkpoint into the checkpoint_dir of your custom config file. If your task is masking the clothings only, then you need to train on your own dataset only. Do not include other datasets since they might include human masks.
2. You may remove them and substitute with your own dataset's directory. The directory should follow the structure of the training datasets that we provide. root/sets/images for input images, root/sets/masks for binary ground truth masks.
3. Just try our download script. https://github.com/plemeri/InSPyReNet/tree/main#easy-download-cake. You'll understand the structure.
4. The more the better. our API is trained with massive composite dataset to achieve better performance. Try your best to get better results.

Thanks.

[plemeri]

Closing due to inactivity.