You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

3.3 KiB

CLI Basics

If you want to train, validate or run inference on models and don't need to make any modifications to the code, using YOLO command line interface is the easiest way to get started.

!!! tip "Syntax" bash yolo task=detect mode=train model=s.yaml epochs=1 ... ... ... ... segment infer s-cls.pt classify val s-seg.pt

The experiment arguments can be overridden directly by pass arg=val covered in the next section. You can run any supported task by setting task and mode in cli. === "Training"

|                    | `task`          | snippet                                                     |
| -----------        | -------------   | ----------------------------------------------------------- |
|  Detection         |  `detect`       | <pre><code>yolo task=detect mode=train       </code></pre>  |
|  Instance Segment  |  `segment`      | <pre><code>yolo task=segment mode=train      </code></pre>  |
|  Classification    |  `classify`     | <pre><code>yolo task=classify mode=train    </code></pre>   |

=== "Inference"

|                    | `task`          | snippet                                                      |
| -----------        | -------------   | ------------------------------------------------------------ |
|  Detection         |  `detect`       | <pre><code>yolo task=detect mode=infer       </code></pre>   |
|  Instance Segment  |  `segment`      | <pre><code>yolo task=segment mode=infer     </code></pre>    |
|  Classification    |  `classify`     | <pre><code>yolo task=classify mode=infer    </code></pre>    |

=== "Validation"

|                    | `task`          | snippet                                                       |
| -----------        | -------------   | ------------------------------------------------------------- |
|  Detection         |  `detect`       | <pre><code>yolo task=detect mode=val        </code></pre>     |
|  Instance Segment  |  `segment`      | <pre><code>yolo task=segment mode=val       </code></pre>     |
|  Classification    |  `classify`     | <pre><code>yolo task=classify mode=val      </code></pre>     |

!!! note "" Note: The arguments don't require '--' prefix. These are reserved for special commands covered later

Overriding default config arguments

All global default arguments can be overriden by simply passing them as arguments in the cli. !!! tip "" === "Syntax" yolo task= ... mode= ... {++ arg=val ++}

=== "Example"
    Perform detection training for `10 epochs` with `learning_rate` of `0.01`
    ```
    yolo task=detect mode=train {++ epochs=10 lr0=0.01 ++}

    ```

Overriding default config file

You can override config file entirely by passing a new file. You can create a copy of default config file in your current working dir as follows:

yolo task=init

You can then use special --cfg name.yaml command to pass the new config file

yolo task=detect mode=train {++ --cfg default.yaml ++}

??? example === "Command" yolo task=init yolo task=detect mode=train --cfg default.yaml === "Result" TODO: add terminal output