File size: 580 Bytes
7ab05dd | 1 2 3 4 5 6 7 8 9 10 11 | # 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 *
|