Accept model=yolov5*u.yaml arguments (#4230)

This commit is contained in:
Glenn Jocher
2023-08-08 14:09:11 +02:00
committed by GitHub
parent 7dcdca3bf4
commit f5fc807fa7
10 changed files with 23 additions and 12 deletions

View File

@ -26,12 +26,10 @@ You can use many of these models directly in the Command Line Interface (CLI) or
## Usage
You can use RT-DETR for object detection tasks using the `ultralytics` pip package. The following is a sample code snippet showing how to use RT-DETR models for training and inference:
This example provides simple inference code for YOLO, SAM and RTDETR models. For more options including handling inference results see [Predict](../modes/predict.md) mode. For using models with additional modes see [Train](../modes/train.md), [Val](../modes/val.md) and [Export](../modes/export.md).
!!! example ""
This example provides simple inference code for YOLO, SAM and RTDETR models. For more options including handling inference results see [Predict](../modes/predict.md) mode. For using models with additional modes see [Train](../modes/train.md), [Val](../modes/val.md) and [Export](../modes/export.md).
=== "Python"
PyTorch pretrained `*.pt` models as well as configuration `*.yaml` files can be passed to the `YOLO()`, `SAM()`, `NAS()` and `RTDETR()` classes to create a model instance in python: