Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

多类别的 image 和 label 数据格式是什么样子的呀? #30

Open
happywt opened this issue Jan 7, 2025 · 4 comments
Open

Comments

@happywt
Copy link

happywt commented Jan 7, 2025

多类别的 image 和 label 数据格式是什么样子的呀?
我有点没看懂那个synapse的预处理,感觉处理出来的png不像是多类别的label呀。

2c2ca5a3-ebd0-4816-b68e-03bfbf9f7a7e

最后用于训练的 image 和 label 是什么格式的呢? 我想训练一个多类别的SDSeg。
训练多类别的SDSeg时,image用的是png吗?label是多类别的mask吗?

@happywt
Copy link
Author

happywt commented Jan 7, 2025

奥奥,好像SDSeg目前只支持二分类。

@lin-tianyu
Copy link
Owner

对于synapse,多分类和二分类用的都是同一套预处理,只是在config里有num_classes的差异(对应的dataset会用不同的方式读取数据,参考ldm/data/synapse.py)。

处理出来的png看起来是全黑的,这是因为这个png里的像素值就是类别的label。假设有5类,则像素值范围只有0,1,2,3,4,5,自然视觉上就看不到东西。

@lin-tianyu
Copy link
Owner

SDSeg的多分类有一个初步版本,但还没能够SOTA。

如果你想试试的话,应该只需要更换一下训练命令中的config文件即可。此外还可以参考:#28#16

@happywt
Copy link
Author

happywt commented Jan 7, 2025

SDSeg的多分类有一个初步版本,但还没能够SOTA。

如果你想试试的话,应该只需要更换一下训练命令中的config文件即可。此外还可以参考:#28#16

好的,谢谢!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants