add yolo v10 and modify pipeline
This commit is contained in:
@ -1,9 +1,29 @@
|
||||
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
||||
|
||||
from .tasks import (BaseModel, ClassificationModel, DetectionModel, SegmentationModel, attempt_load_one_weight,
|
||||
attempt_load_weights, guess_model_scale, guess_model_task, parse_model, torch_safe_load,
|
||||
yaml_model_load)
|
||||
from .tasks import (
|
||||
BaseModel,
|
||||
ClassificationModel,
|
||||
DetectionModel,
|
||||
SegmentationModel,
|
||||
attempt_load_one_weight,
|
||||
attempt_load_weights,
|
||||
guess_model_scale,
|
||||
guess_model_task,
|
||||
parse_model,
|
||||
torch_safe_load,
|
||||
yaml_model_load,
|
||||
)
|
||||
|
||||
__all__ = ('attempt_load_one_weight', 'attempt_load_weights', 'parse_model', 'yaml_model_load', 'guess_model_task',
|
||||
'guess_model_scale', 'torch_safe_load', 'DetectionModel', 'SegmentationModel', 'ClassificationModel',
|
||||
'BaseModel')
|
||||
__all__ = (
|
||||
"attempt_load_one_weight",
|
||||
"attempt_load_weights",
|
||||
"parse_model",
|
||||
"yaml_model_load",
|
||||
"guess_model_task",
|
||||
"guess_model_scale",
|
||||
"torch_safe_load",
|
||||
"DetectionModel",
|
||||
"SegmentationModel",
|
||||
"ClassificationModel",
|
||||
"BaseModel",
|
||||
)
|
||||
|
Reference in New Issue
Block a user