ziqima's picture
initial commit
4893ce0
raw
history blame
573 Bytes
from .builder import build_model
from .default import DefaultSegmentor, DefaultClassifier
# Backbones
from .sparse_unet import *
from .point_transformer import *
from .point_transformer_v2 import *
from .point_transformer_v3 import *
from .stratified_transformer import *
from .spvcnn import *
from .octformer import *
from .oacnns import *
# from .swin3d import *
# Semantic Segmentation
from .context_aware_classifier import *
# Instance Segmentation
from .point_group import *
# Pretraining
from .masked_scene_contrast import *
from .point_prompt_training import *