Update predict and track method return type (#12101)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
parent
65ecf1442e
commit
b7b70c746b
1 changed files with 3 additions and 2 deletions
|
|
@ -8,6 +8,7 @@ import numpy as np
|
|||
import torch
|
||||
|
||||
from ultralytics.cfg import TASK2DATA, get_cfg, get_save_dir
|
||||
from ultralytics.engine.results import Results
|
||||
from ultralytics.hub.utils import HUB_WEB_ROOT
|
||||
from ultralytics.nn.tasks import attempt_load_one_weight, guess_model_task, nn, yaml_model_load
|
||||
from ultralytics.utils import (
|
||||
|
|
@ -399,7 +400,7 @@ class Model(nn.Module):
|
|||
stream: bool = False,
|
||||
predictor=None,
|
||||
**kwargs,
|
||||
) -> list:
|
||||
) -> list[Results]:
|
||||
"""
|
||||
Performs predictions on the given image source using the YOLO model.
|
||||
|
||||
|
|
@ -457,7 +458,7 @@ class Model(nn.Module):
|
|||
stream: bool = False,
|
||||
persist: bool = False,
|
||||
**kwargs,
|
||||
) -> list:
|
||||
) -> list[Results]:
|
||||
"""
|
||||
Conducts object tracking on the specified input source using the registered trackers.
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue