Add Ultralytics ViT Docs (#3230)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

ultralytics/vit/sam/__init__.py

@@ -1,3 +1,5 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
 from .build import build_sam  # noqa
 from .model import SAM  # noqa
 from .modules.prompt_predictor import PromptPredictor  # noqa

ultralytics/vit/sam/amg.py

@@ -1,3 +1,5 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
 import math
 from copy import deepcopy
 from itertools import product

ultralytics/vit/sam/autosize.py

@@ -1,3 +1,5 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
 

ultralytics/vit/sam/build.py

@@ -1,3 +1,5 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
 

ultralytics/vit/sam/model.py

@@ -1,4 +1,7 @@
-# SAM model interface
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+"""
+SAM model interface
+"""
 
 from ultralytics.yolo.cfg import get_cfg
 

ultralytics/vit/sam/modules/__init__.py

@@ -0,0 +1 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license

ultralytics/vit/sam/modules/decoders.py

@@ -1,3 +1,5 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
 from typing import List, Tuple, Type
 
 import torch

ultralytics/vit/sam/modules/encoders.py

@@ -1,3 +1,5 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
 from typing import Any, Optional, Tuple, Type
 
 import numpy as np

ultralytics/vit/sam/modules/mask_generator.py

@@ -1,3 +1,5 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
 

ultralytics/vit/sam/modules/prompt_predictor.py

@@ -1,3 +1,5 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
 from typing import Optional, Tuple
 
 import numpy as np

ultralytics/vit/sam/modules/sam.py

@@ -1,3 +1,5 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
 

ultralytics/vit/sam/modules/transformer.py

@@ -1,3 +1,5 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
 import math
 from typing import Tuple, Type
 

ultralytics/vit/sam/predict.py

@@ -1,3 +1,5 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
 import numpy as np
 import torch