Fix Docs pretty __init__.py URLs (#14550)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
parent
0d059bec0c
commit
b3b3a15086
16 changed files with 49 additions and 22 deletions
|
|
@ -18,6 +18,8 @@ from ultralytics.utils.torch_utils import select_device
|
|||
|
||||
|
||||
class TorchVisionVideoClassifier:
|
||||
"""Classifies videos using pretrained TorchVision models; see https://pytorch.org/vision/stable/."""
|
||||
|
||||
from torchvision.models.video import (
|
||||
MViT_V1_B_Weights,
|
||||
MViT_V2_S_Weights,
|
||||
|
|
@ -133,6 +135,8 @@ class TorchVisionVideoClassifier:
|
|||
|
||||
|
||||
class HuggingFaceVideoClassifier:
|
||||
"""Zero-shot video classifier using Hugging Face models for various devices."""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
labels: List[str],
|
||||
|
|
|
|||
|
|
@ -15,6 +15,8 @@ img_height = 640
|
|||
|
||||
|
||||
class LetterBox:
|
||||
"""Resizes and reshapes images while maintaining aspect ratio by adding padding, suitable for YOLO models."""
|
||||
|
||||
def __init__(
|
||||
self, new_shape=(img_width, img_height), auto=False, scaleFill=False, scaleup=True, center=True, stride=32
|
||||
):
|
||||
|
|
@ -86,6 +88,8 @@ class LetterBox:
|
|||
|
||||
|
||||
class Yolov8TFLite:
|
||||
"""Class for performing object detection using YOLOv8 model converted to TensorFlow Lite format."""
|
||||
|
||||
def __init__(self, tflite_model, input_image, confidence_thres, iou_thres):
|
||||
"""
|
||||
Initializes an instance of the Yolov8TFLite class.
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue