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