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],
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue