mahdichamseddine's picture
Initial release: PT-v3m1 segmentation model (BIMStruct3D pipeline)
7ab05dd verified
Raw History Blame
580 Bytes
# Trimmed for inference-only use: this delivery only ever builds a PT-v3m1
# backbone wrapped in DefaultSegmentorV2, so the other backbones/heads
# (which pull in extra heavy/compiled deps like torchsparse, pointgroup_ops,
# pointops, transformers) aren't imported. Add back whichever of these you
# need if you point `model.backbone.type`/`model.type` at something else.
from .builder import build_model
from .default import DefaultSegmentor, DefaultSegmentorV2, DefaultClassifier
# Backbone actually used by the bundled checkpoint (PT-v3m1)
from .point_transformer_v3 import *