from nndet.models.heads.classifier import ClassifierType, Classifier
from nndet.models.heads.comb import HeadType, AbstractHead
from nndet.models.heads.regressor import RegressorType, Regressor
from nndet.models.heads.segmenter import SegmenterType, Segmenter