Hi, How can I train my dataset?

#13
by llysonylin2012 - opened

hi , I want to train my own clothing dataset , how can i found the train.py or test.py , then i can update the default modal data?
thx~

I'm not sure what you mean by "I want to train my own clothing dataset". If you want to train on your own clothing dataset then use this https://huggingface.co./docs/transformers/main/tasks/semantic_segmentation which is what I followed to train this model. You'll have to adjust some parts but overall most of it is what you need.

HI mattmdjaga, can you confirm if this model can be trained to segment clothes by different part of clothes? such as neckline, sleeve, armhole and body. and the image is a sketch, not a photo

image.png

Yes it can be trained on any type of image, you might need to adjust the output layer size if you're using different classes.

mattmdjaga changed discussion status to closed

Sign up or log in to comment