xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
yolov10/docs/reference/nn/tasks.md
Glenn Jocher bd0f7ecf6f
Docstrings arguments cleanup (#3229) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-17 16:17:07 +02:00

1.4 KiB
Raw Blame History

description keywords
Learn how to work with Ultralytics YOLO Detection, Segmentation & Classification Models, load weights and parse models in PyTorch. neural network, deep learning, computer vision, object detection, image segmentation, image classification, model ensemble, PyTorch

BaseModel

::: ultralytics.nn.tasks.BaseModel



DetectionModel

::: ultralytics.nn.tasks.DetectionModel



SegmentationModel

::: ultralytics.nn.tasks.SegmentationModel



PoseModel

::: ultralytics.nn.tasks.PoseModel



ClassificationModel

::: ultralytics.nn.tasks.ClassificationModel



RTDETRDetectionModel

::: ultralytics.nn.tasks.RTDETRDetectionModel



Ensemble

::: ultralytics.nn.tasks.Ensemble



torch_safe_load

::: ultralytics.nn.tasks.torch_safe_load



attempt_load_weights

::: ultralytics.nn.tasks.attempt_load_weights



attempt_load_one_weight

::: ultralytics.nn.tasks.attempt_load_one_weight



parse_model

::: ultralytics.nn.tasks.parse_model



yaml_model_load

::: ultralytics.nn.tasks.yaml_model_load



guess_model_scale

::: ultralytics.nn.tasks.guess_model_scale



guess_model_task

::: ultralytics.nn.tasks.guess_model_task