search

HUANGLIZI / LViT

[IEEE Transactions on Medical Imaging/TMI] This repo is the official implementation of "LViT: Language meets Vision Transformer in Medical Image Segmentation"
MIT License
298 stars 26 forks source link

How to segment multi-class objects? #16

Closed JAYCHOU2020 closed 1 year ago

JAYCHOU2020 commented 1 year ago

How to segment multi-class objects?

HUANGLIZI commented 1 year ago

You should use the multi-class cross entropy function instead of Binary cross entropy function and set the corresponding num_ class.