From 7f077f7654c1fd8b5c09dd43232f0b3a7d7602f2 Mon Sep 17 00:00:00 2001 From: Ayush Chaurasia Date: Tue, 30 May 2023 22:53:39 +0530 Subject: [PATCH] Import `ultralytics.nn.tasks` as `ultralytics.nn` (#2889) Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Glenn Jocher --- ultralytics/nn/__init__.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/ultralytics/nn/__init__.py b/ultralytics/nn/__init__.py index e69de29..9889b7e 100644 --- a/ultralytics/nn/__init__.py +++ b/ultralytics/nn/__init__.py @@ -0,0 +1,9 @@ +# 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) + +__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')