search

xuebinqin / DIS

This is the repo for our new project Highly Accurate Dichotomous Image Segmentation
Apache License 2.0
2.27k stars 269 forks source link

Add new groups and categories #79

Open Avedena opened 1 year ago

Avedena commented 1 year ago

Hello, I am currently working on an human segmentation dataset and want to contribute a custom model. The main question is, how I can add to the existing model new groups and categories? Its enough to name the training data properly? For example:

22#Human#1#Standing_Front#abc.jpg
22#Human#2#Standing_Back#abc.jpg
22#Human#3#Sitting_Back#abc.jpg

Have you also a list of the v2 groups and categories to match my training data yours? Thanks

juergengunz commented 1 year ago

I did train on a custom human segmentation dataset and the results are good but still some errors. Not sure about categories. Did you acheive any good results and might share the model/dataset?