init commit
This commit is contained in:
27
ultralytics/nn/__init__.py
Normal file
27
ultralytics/nn/__init__.py
Normal file
@@ -0,0 +1,27 @@
|
||||
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
||||
|
||||
from .tasks import (
|
||||
BaseModel,
|
||||
ClassificationModel,
|
||||
DetectionModel,
|
||||
SegmentationModel,
|
||||
guess_model_scale,
|
||||
guess_model_task,
|
||||
load_checkpoint,
|
||||
parse_model,
|
||||
torch_safe_load,
|
||||
yaml_model_load,
|
||||
)
|
||||
|
||||
__all__ = (
|
||||
"load_checkpoint",
|
||||
"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