Qsingle / LearnablePromptSAM

Try to use the SAM-ViT as the backbone to create the learnable prompt for semantic segmentation
Apache License 2.0
77 stars 13 forks source link

num_class #5

Open xiaohongzaizhe opened 1 year ago

xiaohongzaizhe commented 1 year ago

您好 我想问问您是运行什么数据集 num_class 才需要12呢? 如果我只是单一分割血管跟背景是不是 num_class 设置为2就行

thangngoc89 commented 1 year ago

You can take a look at https://github.com/Qsingle/LearnablePromptSAM/issues/3

Qsingle commented 1 year ago

Thanks to @thangngoc89