HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Standardize license headers in Python files (#18696)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Paula Derrenger 2025-01-15 15:36:11 +01:00 • committed by GitHub
parent 3620a07ed7
commit deb87d8ce3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
165 changed files with 169 additions and 169 deletions
Download patch file Download diff file Expand all files Collapse all files

2
docs/build_docs.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
""" """
Automates the building and post-processing of MkDocs documentation, particularly for projects with multilingual content. Automates the building and post-processing of MkDocs documentation, particularly for projects with multilingual content.
It streamlines the workflow for generating localized versions of the documentation and updating HTML links to ensure It streamlines the workflow for generating localized versions of the documentation and updating HTML links to ensure

2
docs/build_reference.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
""" """
Helper file to build Ultralytics Docs reference section. Recursively walks through ultralytics dir and builds an MkDocs Helper file to build Ultralytics Docs reference section. Recursively walks through ultralytics dir and builds an MkDocs
reference section of *.md files composed of classes and functions, and also creates a nav menu for use in mkdocs.yaml. reference section of *.md files composed of classes and functions, and also creates a nav menu for use in mkdocs.yaml.

2
docs/en/models/mobile-sam.md
View file

@ -8,7 +8,7 @@ keywords: MobileSAM, image segmentation, lightweight model, fast segmentation, m
# Mobile Segment Anything (MobileSAM) # Mobile Segment Anything (MobileSAM)
The MobileSAM paper is now available on [arXiv](https://arxiv.org/pdf/2306.14289.pdf). The MobileSAM paper is now available on [arXiv](https://arxiv.org/pdf/2306.14289).
A demonstration of MobileSAM running on a CPU can be accessed at this [demo link](https://huggingface.co/spaces/dhkim2810/MobileSAM). The performance on a Mac i5 CPU takes approximately 3 seconds. On the Hugging Face demo, the interface and lower-performance CPUs contribute to a slower response, but it continues to function effectively. A demonstration of MobileSAM running on a CPU can be accessed at this [demo link](https://huggingface.co/spaces/dhkim2810/MobileSAM). The performance on a Mac i5 CPU takes approximately 3 seconds. On the Hugging Face demo, the interface and lower-performance CPUs contribute to a slower response, but it continues to function effectively.

2
docs/en/yolov5/tutorials/clearml_logging_integration.md
View file

@ -10,7 +10,7 @@ keywords: ClearML, YOLOv5, machine learning, experiment tracking, data versionin
## About ClearML ## About ClearML
[ClearML](https://clear.ml/) is an [open-source](https://github.com/allegroai/clearml) toolbox designed to save you time ⏱️. [ClearML](https://clear.ml/) is an [open-source](https://github.com/clearml/clearml) toolbox designed to save you time ⏱️.
🔨 Track every YOLOv5 training run in the <b>experiment manager</b> 🔨 Track every YOLOv5 training run in the <b>experiment manager</b>

2
examples/RTDETR-ONNXRuntime-Python/main.py
View file

@ -1,3 +1,5 @@
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import argparse import argparse
import cv2 import cv2

2
examples/YOLOv8-Action-Recognition/action_recognition.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import argparse import argparse
import time import time

2
examples/YOLOv8-ONNXRuntime/main.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import argparse import argparse

2
examples/YOLOv8-OpenCV-ONNX-Python/main.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import argparse import argparse

2
examples/YOLOv8-Region-Counter/yolov8_region_counter.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import argparse import argparse
from collections import defaultdict from collections import defaultdict

2
examples/YOLOv8-SAHI-Inference-Video/yolov8_sahi.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import argparse import argparse
from pathlib import Path from pathlib import Path

2
examples/YOLOv8-Segmentation-ONNXRuntime-Python/main.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import argparse import argparse

2
examples/YOLOv8-TFLite-Python/main.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import argparse import argparse
from typing import Tuple, Union from typing import Tuple, Union

2
tests/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from ultralytics.utils import ASSETS, ROOT, WEIGHTS_DIR, checks from ultralytics.utils import ASSETS, ROOT, WEIGHTS_DIR, checks

2
tests/conftest.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import shutil import shutil
from pathlib import Path from pathlib import Path

2
tests/test_cli.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import subprocess import subprocess

2
tests/test_cuda.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from itertools import product from itertools import product
from pathlib import Path from pathlib import Path

2
tests/test_engine.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import sys import sys
from unittest import mock from unittest import mock

2
tests/test_exports.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import shutil import shutil
import uuid import uuid

2
tests/test_integrations.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import contextlib import contextlib
import os import os

2
tests/test_python.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import contextlib import contextlib
import csv import csv

2
tests/test_solutions.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import cv2 import cv2
import pytest import pytest

2
ultralytics/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
__version__ = "8.3.61" __version__ = "8.3.61"

2
ultralytics/cfg/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import shutil import shutil
import subprocess import subprocess

2
ultralytics/data/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from .base import BaseDataset from .base import BaseDataset
from .build import build_dataloader, build_grounding, build_yolo_dataset, load_inference_source from .build import build_dataloader, build_grounding, build_yolo_dataset, load_inference_source

2
ultralytics/data/annotator.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from pathlib import Path from pathlib import Path

2
ultralytics/data/augment.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import math import math
import random import random

2
ultralytics/data/base.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import glob import glob
import math import math

2
ultralytics/data/build.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import os import os
import random import random

2
ultralytics/data/converter.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import json import json
import random import random

2
ultralytics/data/dataset.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import json import json
from collections import defaultdict from collections import defaultdict

2
ultralytics/data/loaders.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import glob import glob
import math import math

2
ultralytics/data/split_dota.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import itertools import itertools
from glob import glob from glob import glob

2
ultralytics/data/utils.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import hashlib import hashlib
import json import json

2
ultralytics/engine/__init__.py
View file

@ -1 +1 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license

2
ultralytics/engine/exporter.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
""" """
Export a YOLO PyTorch model to other formats. TensorFlow exports authored by https://github.com/zldrobit. Export a YOLO PyTorch model to other formats. TensorFlow exports authored by https://github.com/zldrobit.

2
ultralytics/engine/model.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import inspect import inspect
from pathlib import Path from pathlib import Path

2
ultralytics/engine/predictor.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
""" """
Run prediction on images, videos, directories, globs, YouTube, webcam, streams, etc. Run prediction on images, videos, directories, globs, YouTube, webcam, streams, etc.

2
ultralytics/engine/results.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
""" """
Ultralytics Results, Boxes and Masks classes for handling inference results. Ultralytics Results, Boxes and Masks classes for handling inference results.

2
ultralytics/engine/trainer.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
""" """
Train a model on a dataset. Train a model on a dataset.

2
ultralytics/engine/tuner.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
""" """
Module provides functionalities for hyperparameter tuning of the Ultralytics YOLO models for object detection, instance Module provides functionalities for hyperparameter tuning of the Ultralytics YOLO models for object detection, instance
segmentation, image classification, pose estimation, and multi-object tracking. segmentation, image classification, pose estimation, and multi-object tracking.

2
ultralytics/engine/validator.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
""" """
Check a model's accuracy on a test or val split of a dataset. Check a model's accuracy on a test or val split of a dataset.

2
ultralytics/hub/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import requests import requests

2
ultralytics/hub/auth.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import requests import requests

2
ultralytics/hub/google/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import concurrent.futures import concurrent.futures
import statistics import statistics

2
ultralytics/hub/session.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import shutil import shutil
import threading import threading

2
ultralytics/hub/utils.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import os import os
import platform import platform

2
ultralytics/models/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from .fastsam import FastSAM from .fastsam import FastSAM
from .nas import NAS from .nas import NAS

2
ultralytics/models/fastsam/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from .model import FastSAM from .model import FastSAM
from .predict import FastSAMPredictor from .predict import FastSAMPredictor

2
ultralytics/models/fastsam/model.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from pathlib import Path from pathlib import Path

3
ultralytics/models/fastsam/predict.py
View file

@ -1,4 +1,5 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import torch import torch
from PIL import Image from PIL import Image

2
ultralytics/models/fastsam/utils.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
def adjust_bboxes_to_image_border(boxes, image_shape, threshold=20): def adjust_bboxes_to_image_border(boxes, image_shape, threshold=20):

2
ultralytics/models/fastsam/val.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from ultralytics.models.yolo.segment import SegmentationValidator from ultralytics.models.yolo.segment import SegmentationValidator
from ultralytics.utils.metrics import SegmentMetrics from ultralytics.utils.metrics import SegmentMetrics

2
ultralytics/models/nas/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from .model import NAS from .model import NAS
from .predict import NASPredictor from .predict import NASPredictor

2
ultralytics/models/nas/model.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
""" """
YOLO-NAS model interface. YOLO-NAS model interface.

2
ultralytics/models/nas/predict.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import torch import torch

2
ultralytics/models/nas/val.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import torch import torch

2
ultralytics/models/rtdetr/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from .model import RTDETR from .model import RTDETR
from .predict import RTDETRPredictor from .predict import RTDETRPredictor

2
ultralytics/models/rtdetr/model.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
""" """
Interface for Baidu's RT-DETR, a Vision Transformer-based real-time object detector. RT-DETR offers real-time Interface for Baidu's RT-DETR, a Vision Transformer-based real-time object detector. RT-DETR offers real-time
performance and high accuracy, excelling in accelerated backends like CUDA with TensorRT. It features an efficient performance and high accuracy, excelling in accelerated backends like CUDA with TensorRT. It features an efficient

2
ultralytics/models/rtdetr/predict.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import torch import torch

2
ultralytics/models/rtdetr/train.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from copy import copy from copy import copy

2
ultralytics/models/rtdetr/val.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import torch import torch

2
ultralytics/models/sam/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from .model import SAM from .model import SAM
from .predict import Predictor, SAM2Predictor, SAM2VideoPredictor from .predict import Predictor, SAM2Predictor, SAM2VideoPredictor

2
ultralytics/models/sam/amg.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import math import math
from itertools import product from itertools import product

2
ultralytics/models/sam/build.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
# Copyright (c) Meta Platforms, Inc. and affiliates. # Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved. # All rights reserved.

2
ultralytics/models/sam/model.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
""" """
SAM model interface. SAM model interface.

2
ultralytics/models/sam/modules/__init__.py
View file

@ -1 +1 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license

2
ultralytics/models/sam/modules/blocks.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import copy import copy
import math import math

2
ultralytics/models/sam/modules/decoders.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from typing import List, Optional, Tuple, Type from typing import List, Optional, Tuple, Type

2
ultralytics/models/sam/modules/encoders.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from typing import List, Optional, Tuple, Type from typing import List, Optional, Tuple, Type

2
ultralytics/models/sam/modules/memory_attention.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import copy import copy
from typing import Optional from typing import Optional

2
ultralytics/models/sam/modules/sam.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
# Copyright (c) Meta Platforms, Inc. and affiliates. # Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved. # All rights reserved.

2
ultralytics/models/sam/modules/tiny_encoder.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
# -------------------------------------------------------- # --------------------------------------------------------
# TinyViT Model Architecture # TinyViT Model Architecture

2
ultralytics/models/sam/modules/transformer.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import math import math
from typing import Tuple, Type from typing import Tuple, Type

2
ultralytics/models/sam/modules/utils.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from typing import Tuple from typing import Tuple

2
ultralytics/models/sam/predict.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
""" """
Generate predictions using the Segment Anything Model (SAM). Generate predictions using the Segment Anything Model (SAM).

2
ultralytics/models/utils/__init__.py
View file

@ -1 +1 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license

8
ultralytics/models/utils/loss.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import torch import torch
import torch.nn as nn import torch.nn as nn
@ -243,13 +243,11 @@ class DETRLoss(nn.Module):
if len(gt_bboxes): if len(gt_bboxes):
gt_scores[idx] = bbox_iou(pred_bboxes.detach(), gt_bboxes, xywh=True).squeeze(-1) gt_scores[idx] = bbox_iou(pred_bboxes.detach(), gt_bboxes, xywh=True).squeeze(-1)
loss = { return {
**self._get_loss_class(pred_scores, targets, gt_scores, len(gt_bboxes), postfix), **self._get_loss_class(pred_scores, targets, gt_scores, len(gt_bboxes), postfix),
**self._get_loss_bbox(pred_bboxes, gt_bboxes, postfix), **self._get_loss_bbox(pred_bboxes, gt_bboxes, postfix),
# **(self._get_loss_mask(masks, gt_mask, match_indices, postfix) if masks is not None and gt_mask is not None else {})
} }
# if masks is not None and gt_mask is not None:
# loss.update(self._get_loss_mask(masks, gt_mask, match_indices, postfix))
return loss
def forward(self, pred_bboxes, pred_scores, batch, postfix="", **kwargs): def forward(self, pred_bboxes, pred_scores, batch, postfix="", **kwargs):
""" """

2
ultralytics/models/utils/ops.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import torch import torch
import torch.nn as nn import torch.nn as nn

2
ultralytics/models/yolo/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from ultralytics.models.yolo import classify, detect, obb, pose, segment, world from ultralytics.models.yolo import classify, detect, obb, pose, segment, world

2
ultralytics/models/yolo/classify/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from ultralytics.models.yolo.classify.predict import ClassificationPredictor from ultralytics.models.yolo.classify.predict import ClassificationPredictor
from ultralytics.models.yolo.classify.train import ClassificationTrainer from ultralytics.models.yolo.classify.train import ClassificationTrainer

2
ultralytics/models/yolo/classify/predict.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import cv2 import cv2
import torch import torch

2
ultralytics/models/yolo/classify/train.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from copy import copy from copy import copy

2
ultralytics/models/yolo/classify/val.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import torch import torch

2
ultralytics/models/yolo/detect/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from .predict import DetectionPredictor from .predict import DetectionPredictor
from .train import DetectionTrainer from .train import DetectionTrainer

2
ultralytics/models/yolo/detect/predict.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from ultralytics.engine.predictor import BasePredictor from ultralytics.engine.predictor import BasePredictor
from ultralytics.engine.results import Results from ultralytics.engine.results import Results

2
ultralytics/models/yolo/detect/train.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import math import math
import random import random

2
ultralytics/models/yolo/detect/val.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import os import os
from pathlib import Path from pathlib import Path

2
ultralytics/models/yolo/model.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from pathlib import Path from pathlib import Path

2
ultralytics/models/yolo/obb/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from .predict import OBBPredictor from .predict import OBBPredictor
from .train import OBBTrainer from .train import OBBTrainer

2
ultralytics/models/yolo/obb/predict.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
import torch import torch

2
ultralytics/models/yolo/obb/train.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from copy import copy from copy import copy

2
ultralytics/models/yolo/obb/val.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from pathlib import Path from pathlib import Path

2
ultralytics/models/yolo/pose/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from .predict import PosePredictor from .predict import PosePredictor
from .train import PoseTrainer from .train import PoseTrainer

2
ultralytics/models/yolo/pose/predict.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from ultralytics.engine.results import Results from ultralytics.engine.results import Results
from ultralytics.models.yolo.detect.predict import DetectionPredictor from ultralytics.models.yolo.detect.predict import DetectionPredictor

2
ultralytics/models/yolo/pose/train.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from copy import copy from copy import copy

2
ultralytics/models/yolo/pose/val.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from pathlib import Path from pathlib import Path

2
ultralytics/models/yolo/segment/__init__.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from .predict import SegmentationPredictor from .predict import SegmentationPredictor
from .train import SegmentationTrainer from .train import SegmentationTrainer

2
ultralytics/models/yolo/segment/predict.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from ultralytics.engine.results import Results from ultralytics.engine.results import Results
from ultralytics.models.yolo.detect.predict import DetectionPredictor from ultralytics.models.yolo.detect.predict import DetectionPredictor

2
ultralytics/models/yolo/segment/train.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from copy import copy from copy import copy

2
ultralytics/models/yolo/segment/val.py
View file

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from multiprocessing.pool import ThreadPool from multiprocessing.pool import ThreadPool
from pathlib import Path from pathlib import Path

Some files were not shown because too many files have changed in this diff Show more