Visual-Attention-Network / SegNeXt

Official Pytorch implementations for "SegNeXt: Rethinking Convolutional Attention Design for Semantic Segmentation" (NeurIPS 2022)
Apache License 2.0
795 stars 85 forks source link

How do I train a new dataset? #8

Closed roystonlmq closed 2 years ago

roystonlmq commented 2 years ago

Hi, I have a dataset where I am detecting primarily just one object and I wish to create a mask image of that object. The object would appear white against a black background. I have organised it in SegNeXt/data/ images/ annotations/

Currently your README has the following information for training: ./tools/dist_train.sh /path/to/config 8

Any pretrained models for me to edit and retrain on the new dataset?

uyzhang commented 2 years ago

You can refer to mmseg