YYYYYYUUU's picture
Add core reproduction code (binarization layers, PTv3, superpoint ops, min-repro pack)
7b95dc2 verified
Raw
History Blame Contribute Delete
665 Bytes
from .builder import build_model
from .default import DefaultSegmentor, DefaultClassifier
from .modules import PointModule, PointModel
# 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 *
from .sgiformer import *
# Pretraining
from .masked_scene_contrast import *
from .point_prompt_training import *
from .sonata import *