Commit a74dc9a0 authored by chenzk's avatar chenzk
Browse files

v1.0

parents
---
description: Explore the train.py module in Ultralytics YOLO for efficient classification model training. Learn more with examples and detailed code documentation.
keywords: YOLO, Ultralytics, classification, training, machine learning, deep learning, PyTorch, train.py
---
# Reference for `ultralytics/models/yolo/classify/train.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/classify/train.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/classify/train.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/classify/train.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.classify.train.ClassificationTrainer
<br><br>
---
description: Explore the source code and functionalities of the YOLO Classification Validator in Ultralytics for evaluating classification models effectively.
keywords: Ultralytics, YOLO, classification, validation, ClassifyMetrics, ConfusionMatrix, PyTorch, deep learning, model evaluation, AI, machine learning
---
# Reference for `ultralytics/models/yolo/classify/val.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/classify/val.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/classify/val.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/classify/val.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.classify.val.ClassificationValidator
<br><br>
---
description: Explore the Ultralytics YOLO Detection Predictor. Learn how to implement and use the DetectionPredictor class for object detection in Python.
keywords: YOLO, Ultralytics, DetectionPredictor, object detection, Python, machine learning, AI, non_max_suppression
---
# Reference for `ultralytics/models/yolo/detect/predict.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/detect/predict.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/detect/predict.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/detect/predict.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.detect.predict.DetectionPredictor
<br><br>
---
description: Learn about the DetectionTrainer class for training YOLO models on custom datasets. Discover methods, examples, and more.
keywords: Ultralytics, YOLO, DetectionTrainer, training, object detection, machine learning, build dataset, dataloader, detection model
---
# Reference for `ultralytics/models/yolo/detect/train.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/detect/train.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/detect/train.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/detect/train.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.detect.train.DetectionTrainer
<br><br>
---
description: Explore the DetectionValidator class for YOLO models in Ultralytics. Learn validation techniques, metrics, and dataset handling for object detection.
keywords: YOLO validation, detection validation, YOLO metrics, Ultralytics, object detection, machine learning, AI
---
# Reference for `ultralytics/models/yolo/detect/val.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/detect/val.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/detect/val.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/detect/val.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.detect.val.DetectionValidator
<br><br>
---
description: Explore the ultralytics.models.yolo.model module for YOLO object detection. Learn initialization, model mapping, and more.
keywords: YOLO, object detection, Ultralytics, YOLO model, machine learning, Python, model initialization
---
# Reference for `ultralytics/models/yolo/model.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/model.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/model.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/model.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.model.YOLO
<br><br><hr><br>
## ::: ultralytics.models.yolo.model.YOLOWorld
<br><br>
---
description: Learn how to use the Ultralytics YOLO OBBPredictor for oriented bounding box predictions. Enhance your object detection models with ease.
keywords: Ultralytics, YOLO, OBBPredictor, oriented bounding box, object detection, AI, machine learning, PyTorch
---
# Reference for `ultralytics/models/yolo/obb/predict.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/obb/predict.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/obb/predict.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/obb/predict.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.obb.predict.OBBPredictor
<br><br>
---
description: Explore the Ultralytics YOLO OBB Trainer class for efficient training with Oriented Bounding Box models. Learn with examples and method details.
keywords: Ultralytics, YOLO, OBB Trainer, Oriented Bounding Box, Machine Learning, Training, AI
---
# Reference for `ultralytics/models/yolo/obb/train.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/obb/train.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/obb/train.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/obb/train.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.obb.train.OBBTrainer
<br><br>
---
description: Explore the OBBValidator for YOLO, an advanced class for oriented bounding boxes (OBB). Learn initialization, processes, and evaluation methods.
keywords: Ultralytics, YOLO, OBBValidator, Oriented Bounding Boxes, DetectionValidator, validation, Python, deep learning
---
# Reference for `ultralytics/models/yolo/obb/val.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/obb/val.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/obb/val.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/obb/val.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.obb.val.OBBValidator
<br><br>
---
description: Learn about the PosePredictor class for YOLO model predictions on pose data. Get setup instructions, example usage, and implementation details.
keywords: YOLO, Pose Prediction, Ultralytics, PosePredictor, YOLOv8, Machine Learning, Deep Learning, Python, AI Models
---
# Reference for `ultralytics/models/yolo/pose/predict.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/pose/predict.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/pose/predict.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/pose/predict.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.pose.predict.PosePredictor
<br><br>
---
description: Explore the PoseTrainer class for training pose models using YOLO from Ultralytics. Includes initialization, model configuration, and plotting methods.
keywords: PoseTrainer, YOLO, Ultralytics, pose models, training, model configuration, deep learning, machine learning, pose estimation
---
# Reference for `ultralytics/models/yolo/pose/train.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/pose/train.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/pose/train.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/pose/train.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.pose.train.PoseTrainer
<br><br>
---
description: Explore the PoseValidator class for YOLO models. Learn how to extend DetectionValidator for pose validation with example code and detailed methods.
keywords: Ultralytics, YOLO, PoseValidator, pose validation, machine learning, object detection, keypoints, python code, AI, deep learning
---
# Reference for `ultralytics/models/yolo/pose/val.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/pose/val.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/pose/val.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/pose/val.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.pose.val.PoseValidator
<br><br>
---
description: Understand the SegmentationPredictor class for segmentation-based predictions using YOLO. Learn more about its implementation and example usage.
keywords: YOLO, SegmentationPredictor, machine learning, computer vision, object detection, Ultralytics, prediction, model, non-max suppression
---
# Reference for `ultralytics/models/yolo/segment/predict.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/segment/predict.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/segment/predict.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/segment/predict.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.segment.predict.SegmentationPredictor
<br><br>
---
description: Learn how to train YOLO models for segmentation tasks with Ultralytics. Explore the SegmentationTrainer class and its functionalities.
keywords: YOLO, segmentation, train, Ultralytics, SegmentationTrainer, Python, machine learning, deep learning, tutorials
---
# Reference for `ultralytics/models/yolo/segment/train.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/segment/train.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/segment/train.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/segment/train.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.segment.train.SegmentationTrainer
<br><br>
---
description: Explore the YOLO Segmentation Validator module for validating segment models. Understand its usage, metrics, and implementation within the Ultralytics framework.
keywords: YOLO, segmentation, validator, Ultralytics, model validation, machine learning, deep learning, AI, computer vision
---
# Reference for `ultralytics/models/yolo/segment/val.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/segment/val.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/segment/val.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/segment/val.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.segment.val.SegmentationValidator
<br><br>
---
description: Learn how to train a World Model with Ultralytics YOLO using advanced techniques and customizable options for optimal performance.
keywords: Ultralytics, YOLO, World Model, training, deep learning, computer vision, AI, machine learning, tutorial
---
# Reference for `ultralytics/models/yolo/world/train.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/world/train.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/world/train.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/world/train.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.world.train.WorldTrainer
<br><br><hr><br>
## ::: ultralytics.models.yolo.world.train.on_pretrain_routine_end
<br><br>
---
description: Explore the WorldTrainerFromScratch in YOLO for open-set datasets. Learn how to build, train, and evaluate models efficiently.
keywords: YOLO, WorldTrainer, open-set datasets, training, evaluation, build dataset, YOLO World, machine learning
---
# Reference for `ultralytics/models/yolo/world/train_world.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/world/train_world.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/world/train_world.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/world/train_world.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.models.yolo.world.train_world.WorldTrainerFromScratch
<br><br>
---
description: Get to know more about Ultralytics nn.autobackend.check_class_names functionality. Optimize your YOLO models seamlessly.
keywords: Ultralytics, AutoBackend, check_class_names, YOLO, YOLO models, optimization
---
# Reference for `ultralytics/nn/autobackend.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/nn/autobackend.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/nn/autobackend.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/nn/autobackend.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.nn.autobackend.AutoBackend
<br><br><hr><br>
## ::: ultralytics.nn.autobackend.check_class_names
<br><br><hr><br>
## ::: ultralytics.nn.autobackend.default_class_names
<br><br>
---
description: Explore activation functions in Ultralytics, including the Unified activation function and other custom implementations for neural networks.
keywords: ultralytics, activation functions, neural networks, Unified activation, AGLU, SiLU, ReLU, PyTorch, deep learning, custom activations
---
# Reference for `ultralytics/nn/modules/activation.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/nn/modules/activation.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/nn/modules/activation.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/nn/modules/activation.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.nn.modules.activation.AGLU
<br><br>
---
description: Explore detailed documentation of block modules in Ultralytics, available for deep learning tasks. Contribute and improve the codebase!.
keywords: Ultralytics, YOLO, neural networks, block modules, DFL, Proto, HGStem, HGBlock, SPP, SPPF, C1, C2, C2f, C3, C3x, RepC3, C3TR, C3Ghost, GhostBottleneck, Bottleneck, BottleneckCSP, ResNetBlock, MaxSigmoidAttnBlock, ImagePoolingAttn, ContrastiveHead, RepBottleneck, RepCSP, RepNCSPELAN4, ADown, SPPELAN, Silence, CBLinear, CBFuse
---
# Reference for `ultralytics/nn/modules/block.py`
!!! note
This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/nn/modules/block.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/nn/modules/block.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/nn/modules/block.py) 🛠️. Thank you 🙏!
<br>
## ::: ultralytics.nn.modules.block.DFL
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.Proto
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.HGStem
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.HGBlock
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.SPP
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.SPPF
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C1
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C2
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C2f
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C3
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C3x
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.RepC3
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C3TR
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C3Ghost
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.GhostBottleneck
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.Bottleneck
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.BottleneckCSP
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.ResNetBlock
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.ResNetLayer
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.MaxSigmoidAttnBlock
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C2fAttn
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.ImagePoolingAttn
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.ContrastiveHead
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.BNContrastiveHead
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.RepBottleneck
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.RepCSP
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.RepNCSPELAN4
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.ELAN1
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.AConv
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.ADown
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.SPPELAN
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.CBLinear
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.CBFuse
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C3f
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C3k2
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C3k
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.RepVGGDW
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.CIB
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C2fCIB
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.Attention
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.PSABlock
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.PSA
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C2PSA
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.C2fPSA
<br><br><hr><br>
## ::: ultralytics.nn.modules.block.SCDown
<br><br>
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment