search

Visual-Attention-Network / SegNeXt

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

Hello, can you support binary segmentation dataset? 0 or 255(eg: Medical Image Segmentation) #15

Closed zwb170010 closed 2 years ago

MenghaoGuo commented 2 years ago

Yes, we have plan to expand our model for medical image segmentation.

zwb170010 commented 2 years ago

Yes, we have plan to expand our model for medical image segmentation.

Is the current git not supported? I tried to process the medical image, but I encountered some problems when calculating the loss.

uyzhang commented 2 years ago

You can convert 255 to 1, and then perform binary classification. It should also be effective.

zwb170010 commented 2 years ago

You can convert 255 to 1, and then perform binary classification. It should also be effective.

Thanks.