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 <glenn.jocher@ultralytics.com>
2 years ago · 7f077f7654
parent 50c9731808
commit 7f077f7654
1 changed files with 9 additions and 0 deletions
--- 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')