openpath / OpenPath /dinov2 /eval /segmentation /models /backbones /vision_transformer.py
taejoon89's picture
Upload folder using huggingface_hub
62a0e4e verified
Raw
History Blame Contribute Delete
476 Bytes
# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This source code is licensed under the Apache License, Version 2.0
# found in the LICENSE file in the root directory of this source tree.
from mmcv.runner import BaseModule
from mmseg.models.builder import BACKBONES
@BACKBONES.register_module()
class DinoVisionTransformer(BaseModule):
"""Vision Transformer."""
def __init__(
self,
*args,
**kwargs,
):
super().__init__()