Add missing single-line docstrings (#8362)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Johannes Kaisinger <johannes.kaisinger@gmail.com>
pull/8395/head
Glenn Jocher 1 year ago committed by GitHub
parent 2945cfc6ef
commit 8c28e0c3fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      docs/build_docs.py
  2. 2
      tests/test_python.py
  3. 2
      ultralytics/models/yolo/obb/train.py

@ -115,7 +115,7 @@ def update_subdir_edit_links(subdir="", docs_url=""):
def main():
# Build the docs
"""Builds docs, updates titles and edit links, and prints local server command."""
build_docs()
# Update titles

@ -555,6 +555,7 @@ def test_hub():
@pytest.fixture
def image():
"""Loads an image from a predefined source using OpenCV."""
return cv2.imread(str(SOURCE))
@ -568,6 +569,7 @@ def image():
],
)
def test_classify_transforms_train(image, auto_augment, erasing, force_color_jitter):
"""Tests classification transforms during training with various augmentation settings."""
import torchvision.transforms as T
from ultralytics.data.augment import classify_augmentations

@ -15,7 +15,7 @@ class OBBTrainer(yolo.detect.DetectionTrainer):
```python
from ultralytics.models.yolo.obb import OBBTrainer
args = dict(model='yolov8n-seg.pt', data='coco8-seg.yaml', epochs=3)
args = dict(model='yolov8n-obb.pt', data='dota8.yaml', epochs=3)
trainer = OBBTrainer(overrides=args)
trainer.train()
```

Loading…
Cancel
Save