search

MontaEllis / Pytorch-Medical-Segmentation

This repository is an unoffical PyTorch implementation of Medical segmentation in 2D and 3D.
MIT License
857 stars 197 forks source link

如何让模型可以处理大于1的标签？ #34

Closed Fuzhiyuan closed 2 years ago

Fuzhiyuan commented 2 years ago

现在的模型训练后只是把脊椎的所有节都分成一个颜色，我的训练集标签是每相邻节骨头都是不一样的颜色，我也需要这样的推理效果，请问应该怎么处理啊？

JohnMasoner commented 2 years ago

Your problem maybe a multi-class task, but your label is binary classification label.

MontaEllis commented 2 years ago

我们即将对代码进行一次大的更新，您的需求我们会考虑进去。