How to display the specified classes with custom_segmentation?

ayoolaolafenwa / PixelLib

Visit PixelLib's official documentation https://pixellib.readthedocs.io/en/latest/

MIT License

1.04k stars 267 forks source link

How to display the specified classes with custom_segmentation? #145

Open dapsjj opened 2 years ago

dapsjj commented 2 years ago

I made and trained my own dataset, but I found that there is no segment_target_classes parameter when using custom_segmentation to predict.
How to use segment_target_classes in custom_segmentation?