43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
"""
|
|
VGGT (Visual Geometry Grounded Transformer) Package
|
|
Facebook Research
|
|
"""
|
|
|
|
__version__ = "1.0.0"
|
|
__author__ = "Facebook Research"
|
|
|
|
# 导入核心模块，确保可以通过 from vggt.models.vggt import VGGT 访问
|
|
try:
|
|
from .models.vggt import VGGT
|
|
from .models.aggregator import Aggregator
|
|
|
|
# 导入头部模块
|
|
from .heads.camera_head import CameraHead
|
|
from .heads.dpt_head import DPTHead
|
|
from .heads.track_head import TrackHead
|
|
|
|
# 导入工具模块
|
|
from . import utils
|
|
from . import layers
|
|
|
|
__all__ = [
|
|
'VGGT',
|
|
'Aggregator',
|
|
'CameraHead',
|
|
'DPTHead',
|
|
'TrackHead',
|
|
'utils',
|
|
'layers'
|
|
]
|
|
|
|
except ImportError as e:
|
|
# 如果导入失败，仍然允许模块被导入，但会在使用时报错
|
|
import warnings
|
|
warnings.warn(f"VGGT模块导入部分失败: {e}")
|
|
|
|
# 定义空的类以避免导入错误
|
|
class VGGT:
|
|
def __init__(self, *args, **kwargs):
|
|
raise ImportError("VGGT模型导入失败，请检查依赖")
|
|
|
|
__all__ = ['VGGT'] |