Miao Zheng
Miao Zheng
The shape of the seg map must be [H, W], so the pixel value is 0 or 1 rather than [0,0,0] or [1,1,1]
`workers_per_gpu` is the `num_workers` for pytorch dataloader, and I think `workers_per_gpu=16` is a little high for your CPU, and I suggest you set the `num_workers` lower
I see, and suggest you try to set`workers_per_gpu=0` and `persistent_workers=False` to debug.
You might try to double-check the Config of SETR, and there is no `class_weight` in model cfg. ref: https://github.com/open-mmlab/mmsegmentation/blob/master/configs/_base_/models/setr_mla.py
Would you mind sharing more information about your dataset and how you implement it?
Could you please check the num_class in your model configuration?
Thanks for your bug reporting. We will fix it asap. @xiexinch please have a look.
would you like to provide more information about your issue
I would like to suggest you double-check your dataset, and there might be something wrong with the ground truth.
We are planning these models indeed. However, it won't be finished in a short time due to the lack of developers. Collaborations are appreciated If you are interested in it.