ultralytics 8.2.81__py3-none-any.whl → 8.2.83__py3-none-any.whl

ultralytics/engine/results.py

@@ -143,7 +143,7 @@ class BaseTensor(SimpleClass):
 
         Examples:
             >>> base_tensor = BaseTensor(torch.randn(3, 4), orig_shape=(480, 640))
-            >>> cuda_tensor = base_tensor.to('cuda')
+            >>> cuda_tensor = base_tensor.to("cuda")
             >>> float16_tensor = base_tensor.to(dtype=torch.float16)
         """
         return self.__class__(torch.as_tensor(self.data).to(*args, **kwargs), self.orig_shape)
@@ -223,7 +223,7 @@ class Results(SimpleClass):
         >>> for result in results:
         ...     print(result.boxes)  # Print detection boxes
         ...     result.show()  # Display the annotated image
-        ...     result.save(filename='result.jpg')  # Save annotated image
+        ...     result.save(filename="result.jpg")  # Save annotated image
     """
 
     def __init__(
@@ -280,7 +280,7 @@ class Results(SimpleClass):
             (Results): A new Results object containing the specified subset of inference results.
 
         Examples:
-            >>> results = model('path/to/image.jpg')  # Perform inference
+            >>> results = model("path/to/image.jpg")  # Perform inference
             >>> single_result = results[0]  # Get the first result
             >>> subset_results = results[1:4]  # Get a slice of results
         """
@@ -319,7 +319,7 @@ class Results(SimpleClass):
             obb (torch.Tensor | None): A tensor of shape (N, 5) containing oriented bounding box coordinates.
 
         Examples:
-            >>> results = model('image.jpg')
+            >>> results = model("image.jpg")
             >>> new_boxes = torch.tensor([[100, 100, 200, 200, 0.9, 0]])
             >>> results[0].update(boxes=new_boxes)
         """
@@ -370,7 +370,7 @@ class Results(SimpleClass):
             (Results): A new Results object with all tensor attributes on CPU memory.
 
         Examples:
-            >>> results = model('path/to/image.jpg')  # Perform inference
+            >>> results = model("path/to/image.jpg")  # Perform inference
             >>> cpu_result = results[0].cpu()  # Move the first result to CPU
             >>> print(cpu_result.boxes.device)  # Output: cpu
         """
@@ -384,7 +384,7 @@ class Results(SimpleClass):
             (Results): A new Results object with all tensors converted to numpy arrays.
 
         Examples:
-            >>> results = model('path/to/image.jpg')
+            >>> results = model("path/to/image.jpg")
             >>> numpy_result = results[0].numpy()
             >>> type(numpy_result.boxes.data)
             <class 'numpy.ndarray'>
@@ -488,7 +488,7 @@ class Results(SimpleClass):
             (np.ndarray): Annotated image as a numpy array.
 
         Examples:
-            >>> results = model('image.jpg')
+            >>> results = model("image.jpg")
             >>> for result in results:
             ...     im = result.plot()
             ...     im.show()
@@ -578,7 +578,7 @@ class Results(SimpleClass):
             **kwargs (Any): Arbitrary keyword arguments to be passed to the `plot()` method.
 
         Examples:
-            >>> results = model('path/to/image.jpg')
+            >>> results = model("path/to/image.jpg")
             >>> results[0].show()  # Display the first result
             >>> for result in results:
             ...     result.show()  # Display all results
@@ -599,12 +599,12 @@ class Results(SimpleClass):
             **kwargs (Any): Arbitrary keyword arguments to be passed to the `plot` method.
 
         Examples:
-            >>> results = model('path/to/image.jpg')
+            >>> results = model("path/to/image.jpg")
             >>> for result in results:
-            ...     result.save('annotated_image.jpg')
+            ...     result.save("annotated_image.jpg")
             >>> # Or with custom plot arguments
             >>> for result in results:
-            ...     result.save('annotated_image.jpg', conf=False, line_width=2)
+            ...     result.save("annotated_image.jpg", conf=False, line_width=2)
         """
         if not filename:
             filename = f"results_{Path(self.path).name}"
@@ -623,7 +623,7 @@ class Results(SimpleClass):
                 number of detections per class. For classification tasks, it includes the top 5 class probabilities.
 
         Examples:
-            >>> results = model('path/to/image.jpg')
+            >>> results = model("path/to/image.jpg")
             >>> for result in results:
             ...     print(result.verbose())
             2 persons, 1 car, 3 traffic lights,
@@ -660,7 +660,7 @@ class Results(SimpleClass):
 
         Examples:
             >>> from ultralytics import YOLO
-            >>> model = YOLO('yolov8n.pt')
+            >>> model = YOLO("yolov8n.pt")
             >>> results = model("path/to/image.jpg")
             >>> for result in results:
             ...     result.save_txt("output.txt")
@@ -757,7 +757,7 @@ class Results(SimpleClass):
                 task type (classification or detection) and available information (boxes, masks, keypoints).
 
         Examples:
-            >>> results = model('image.jpg')
+            >>> results = model("image.jpg")
             >>> summary = results[0].summary()
             >>> print(summary)
         """
@@ -919,7 +919,7 @@ class Boxes(BaseTensor):
                 coordinates in [x1, y1, x2, y2] format, where n is the number of boxes.
 
         Examples:
-            >>> results = model('image.jpg')
+            >>> results = model("image.jpg")
             >>> boxes = results[0].boxes
             >>> xyxy = boxes.xyxy
             >>> print(xyxy)
@@ -953,7 +953,7 @@ class Boxes(BaseTensor):
                 The shape is (N,), where N is the number of boxes.
 
         Examples:
-            >>> results = model('image.jpg')
+            >>> results = model("image.jpg")
             >>> boxes = results[0].boxes
             >>> class_ids = boxes.cls
             >>> print(class_ids)  # tensor([0., 2., 1.])
@@ -970,7 +970,7 @@ class Boxes(BaseTensor):
                 otherwise None. Shape is (N,) where N is the number of boxes.
 
         Examples:
-            >>> results = model.track('path/to/video.mp4')
+            >>> results = model.track("path/to/video.mp4")
             >>> for result in results:
             ...     boxes = result.boxes
             ...     if boxes.is_track:
@@ -1116,7 +1116,7 @@ class Masks(BaseTensor):
                 mask contour.
 
         Examples:
-            >>> results = model('image.jpg')
+            >>> results = model("image.jpg")
             >>> masks = results[0].masks
             >>> normalized_coords = masks.xyn
             >>> print(normalized_coords[0])  # Normalized coordinates of the first mask
@@ -1141,7 +1141,7 @@ class Masks(BaseTensor):
                 number of points in the segment.
 
         Examples:
-            >>> results = model('image.jpg')
+            >>> results = model("image.jpg")
             >>> masks = results[0].masks
             >>> xy_coords = masks.xy
             >>> print(len(xy_coords))  # Number of masks
@@ -1223,7 +1223,7 @@ class Keypoints(BaseTensor):
                 the number of detections and K is the number of keypoints per detection.
 
         Examples:
-            >>> results = model('image.jpg')
+            >>> results = model("image.jpg")
             >>> keypoints = results[0].keypoints
             >>> xy = keypoints.xy
             >>> print(xy.shape)  # (N, K, 2)
@@ -1388,7 +1388,7 @@ class Probs(BaseTensor):
             (torch.Tensor | numpy.ndarray): A tensor containing the confidence score of the top 1 class.
 
         Examples:
-            >>> results = model('image.jpg')  # classify an image
+            >>> results = model("image.jpg")  # classify an image
             >>> probs = results[0].probs  # get classification probabilities
             >>> top1_confidence = probs.top1conf  # get confidence of top 1 class
             >>> print(f"Top 1 class confidence: {top1_confidence.item():.4f}")
@@ -1410,7 +1410,7 @@ class Probs(BaseTensor):
                 top 5 predicted classes, sorted in descending order of probability.
 
         Examples:
-            >>> results = model('image.jpg')
+            >>> results = model("image.jpg")
             >>> probs = results[0].probs
             >>> top5_conf = probs.top5conf
             >>> print(top5_conf)  # Prints confidence scores for top 5 classes
@@ -1497,7 +1497,7 @@ class OBB(BaseTensor):
                 [x_center, y_center, width, height, rotation]. The shape is (N, 5) where N is the number of boxes.
 
         Examples:
-            >>> results = model('image.jpg')
+            >>> results = model("image.jpg")
             >>> obb = results[0].obb
             >>> xywhr = obb.xywhr
             >>> print(xywhr.shape)
@@ -1518,7 +1518,7 @@ class OBB(BaseTensor):
                 for N detections, where each score is in the range [0, 1].
 
         Examples:
-            >>> results = model('image.jpg')
+            >>> results = model("image.jpg")
             >>> obb_result = results[0].obb
             >>> confidence_scores = obb_result.conf
             >>> print(confidence_scores)
@@ -1535,7 +1535,7 @@ class OBB(BaseTensor):
                 bounding box. The shape is (N,), where N is the number of boxes.
 
         Examples:
-            >>> results = model('image.jpg')
+            >>> results = model("image.jpg")
             >>> result = results[0]
             >>> obb = result.obb
             >>> class_values = obb.cls
@@ -1553,7 +1553,7 @@ class OBB(BaseTensor):
                 oriented bounding box. Returns None if tracking IDs are not available.
 
         Examples:
-            >>> results = model('image.jpg', tracker=True)  # Run inference with tracking
+            >>> results = model("image.jpg", tracker=True)  # Run inference with tracking
             >>> for result in results:
             ...     if result.obb is not None:
             ...         track_ids = result.obb.id
@@ -1620,8 +1620,8 @@ class OBB(BaseTensor):
         Examples:
             >>> import torch
             >>> from ultralytics import YOLO
-            >>> model = YOLO('yolov8n-obb.pt')
-            >>> results = model('path/to/image.jpg')
+            >>> model = YOLO("yolov8n-obb.pt")
+            >>> results = model("path/to/image.jpg")
             >>> for result in results:
             ...     obb = result.obb
             ...     if obb is not None:

ultralytics/engine/trainer.py

@@ -56,8 +56,6 @@ from ultralytics.utils.torch_utils import (
 
 class BaseTrainer:
     """
-    BaseTrainer.
-
     A base class for creating trainers.
 
     Attributes:
@@ -230,7 +228,6 @@ class BaseTrainer:
 
     def _setup_train(self, world_size):
         """Builds dataloaders and optimizer on correct rank process."""
-
         # Model
         self.run_callbacks("on_pretrain_routine_start")
         ckpt = self.setup_model()
@@ -478,12 +475,16 @@ class BaseTrainer:
         torch.cuda.empty_cache()
         self.run_callbacks("teardown")
 
+    def read_results_csv(self):
+        """Read results.csv into a dict using pandas."""
+        import pandas as pd  # scope for faster 'import ultralytics'
+
+        return {k.strip(): v for k, v in pd.read_csv(self.csv).to_dict(orient="list").items()}
+
     def save_model(self):
         """Save model training checkpoints with additional metadata."""
         import io
 
-        import pandas as pd  # scope for faster 'import ultralytics'
-
         # Serialize ckpt to a byte buffer once (faster than repeated torch.save() calls)
         buffer = io.BytesIO()
         torch.save(
@@ -496,7 +497,7 @@ class BaseTrainer:
                 "optimizer": convert_optimizer_state_dict_to_fp16(deepcopy(self.optimizer.state_dict())),
                 "train_args": vars(self.args),  # save as dict
                 "train_metrics": {**self.metrics, **{"fitness": self.fitness}},
-                "train_results": {k.strip(): v for k, v in pd.read_csv(self.csv).to_dict(orient="list").items()},
+                "train_results": self.read_results_csv(),
                 "date": datetime.now().isoformat(),
                 "version": __version__,
                 "license": "AGPL-3.0 (https://ultralytics.com/license)",
@@ -636,7 +637,7 @@ class BaseTrainer:
         pass
 
     def on_plot(self, name, data=None):
-        """Registers plots (e.g. to be consumed in callbacks)"""
+        """Registers plots (e.g. to be consumed in callbacks)."""
         path = Path(name)
         self.plots[path] = {"data": data, "timestamp": time.time()}
 
@@ -646,6 +647,9 @@ class BaseTrainer:
             if f.exists():
                 strip_optimizer(f)  # strip optimizers
                 if f is self.best:
+                    if self.last.is_file():  # update best.pt train_metrics from last.pt
+                        k = "train_results"
+                        torch.save({**torch.load(self.best), **{k: torch.load(self.last)[k]}}, self.best)
                     LOGGER.info(f"\nValidating {f}...")
                     self.validator.args.plots = self.args.plots
                     self.metrics = self.validator(model=f)
@@ -732,7 +736,6 @@ class BaseTrainer:
         Returns:
             (torch.optim.Optimizer): The constructed optimizer.
         """
-
         g = [], [], []  # optimizer parameter groups
         bn = tuple(v for k, v in nn.__dict__.items() if "Norm" in k)  # normalization layers, i.e. BatchNorm2d()
         if name == "auto":

ultralytics/engine/tuner.py

@@ -1,7 +1,7 @@
 # Ultralytics YOLO 🚀, AGPL-3.0 license
 """
-This module provides functionalities for hyperparameter tuning of the Ultralytics YOLO models for object detection,
-instance segmentation, image classification, pose estimation, and multi-object tracking.
+Module provides functionalities for hyperparameter tuning of the Ultralytics YOLO models for object detection, instance
+segmentation, image classification, pose estimation, and multi-object tracking.
 
 Hyperparameter tuning is the process of systematically searching for the optimal set of hyperparameters
 that yield the best model performance. This is particularly crucial in deep learning models like YOLO,
@@ -12,8 +12,8 @@ Example:
     ```python
     from ultralytics import YOLO
 
-    model = YOLO('yolov8n.pt')
-    model.tune(data='coco8.yaml', epochs=10, iterations=300, optimizer='AdamW', plots=False, save=False, val=False)
+    model = YOLO("yolov8n.pt")
+    model.tune(data="coco8.yaml", epochs=10, iterations=300, optimizer="AdamW", plots=False, save=False, val=False)
     ```
 """
 
@@ -54,15 +54,15 @@ class Tuner:
         ```python
         from ultralytics import YOLO
 
-        model = YOLO('yolov8n.pt')
-        model.tune(data='coco8.yaml', epochs=10, iterations=300, optimizer='AdamW', plots=False, save=False, val=False)
+        model = YOLO("yolov8n.pt")
+        model.tune(data="coco8.yaml", epochs=10, iterations=300, optimizer="AdamW", plots=False, save=False, val=False)
         ```
 
         Tune with custom search space.
         ```python
         from ultralytics import YOLO
 
-        model = YOLO('yolov8n.pt')
+        model = YOLO("yolov8n.pt")
         model.tune(space={key1: val1, key2: val2})  # custom search space dictionary
         ```
     """
@@ -176,7 +176,6 @@ class Tuner:
            The method utilizes the `self.tune_csv` Path object to read and log hyperparameters and fitness scores.
            Ensure this path is set correctly in the Tuner instance.
         """
-
         t0 = time.time()
         best_save_dir, best_metrics = None, None
         (self.tune_dir / "weights").mkdir(parents=True, exist_ok=True)

ultralytics/engine/validator.py

@@ -104,9 +104,7 @@ class BaseValidator:
 
     @smart_inference_mode()
     def __call__(self, trainer=None, model=None):
-        """Supports validation of a pre-trained model if passed or a model being trained if trainer is passed (trainer
-        gets priority).
-        """
+        """Executes validation process, running inference on dataloader and computing performance metrics."""
         self.training = trainer is not None
         augment = self.args.augment and (not self.training)
         if self.training:
@@ -280,7 +278,7 @@ class BaseValidator:
         return batch
 
     def postprocess(self, preds):
-        """Describes and summarizes the purpose of 'postprocess()' but no details mentioned."""
+        """Preprocesses the predictions."""
         return preds
 
     def init_metrics(self, model):
@@ -317,7 +315,7 @@ class BaseValidator:
         return []
 
     def on_plot(self, name, data=None):
-        """Registers plots (e.g. to be consumed in callbacks)"""
+        """Registers plots (e.g. to be consumed in callbacks)."""
         self.plots[Path(name)] = {"data": data, "timestamp": time.time()}
 
     # TODO: may need to put these following functions into callback

ultralytics/hub/__init__.py

@@ -136,11 +136,11 @@ def check_dataset(path: str, task: str) -> None:
         ```python
         from ultralytics.hub import check_dataset
 
-        check_dataset('path/to/coco8.zip', task='detect')  # detect dataset
-        check_dataset('path/to/coco8-seg.zip', task='segment')  # segment dataset
-        check_dataset('path/to/coco8-pose.zip', task='pose')  # pose dataset
-        check_dataset('path/to/dota8.zip', task='obb')  # OBB dataset
-        check_dataset('path/to/imagenet10.zip', task='classify')  # classification dataset
+        check_dataset("path/to/coco8.zip", task="detect")  # detect dataset
+        check_dataset("path/to/coco8-seg.zip", task="segment")  # segment dataset
+        check_dataset("path/to/coco8-pose.zip", task="pose")  # pose dataset
+        check_dataset("path/to/dota8.zip", task="obb")  # OBB dataset
+        check_dataset("path/to/imagenet10.zip", task="classify")  # classification dataset
         ```
     """
     HUBDatasetStats(path=path, task=task).get_json()

ultralytics/hub/auth.py

@@ -27,10 +27,14 @@ class Auth:
 
     def __init__(self, api_key="", verbose=False):
         """
-        Initialize the Auth class with an optional API key.
+        Initialize Auth class and authenticate user.
+
+        Handles API key validation, Google Colab authentication, and new key requests. Updates SETTINGS upon successful
+        authentication.
 
         Args:
-            api_key (str, optional): May be an API key or a combination API key and model ID, i.e. key_id
+            api_key (str): API key or combined key_id format.
+            verbose (bool): Enable verbose logging.
         """
         # Split the input API key in case it contains a combined key_model and keep only the API key part
         api_key = api_key.split("_")[0]

ultralytics/hub/session.py

@@ -159,7 +159,6 @@ class HUBTrainingSession:
         Raises:
             HUBModelError: If the identifier format is not recognized.
         """
-
         # Initialize variables
         api_key, model_id, filename = None, None, None
 
@@ -200,7 +199,6 @@ class HUBTrainingSession:
             ValueError: If the model is already trained, if required dataset information is missing, or if there are
                 issues with the provided training arguments.
         """
-
         if self.model.is_resumable():
             # Model has saved weights
             self.train_args = {"data": self.model.get_dataset_url(), "resume": True}
@@ -276,7 +274,7 @@ class HUBTrainingSession:
 
             # if request related to metrics upload and exceed retries
             if response is None and kwargs.get("metrics"):
-                self.metrics_upload_failed_queue.update(kwargs.get("metrics", None))
+                self.metrics_upload_failed_queue.update(kwargs.get("metrics"))
 
             return response
 
@@ -350,10 +348,10 @@ class HUBTrainingSession:
             last = weights.with_name("last" + weights.suffix)
             if final and last.is_file():
                 LOGGER.warning(
-                    f"{PREFIX} ARNING ⚠️ Model 'best.pt' not found, copying 'last.pt' to 'best.pt' and uploading. "
+                    f"{PREFIX} WARNING ⚠️ Model 'best.pt' not found, copying 'last.pt' to 'best.pt' and uploading. "
                     "This often happens when resuming training in transient environments like Google Colab. "
                     "For more reliable training, consider using Ultralytics HUB Cloud. "
-                    "Learn more at https://docs.ultralytics.com/hub/cloud-training/."
+                    "Learn more at https://docs.ultralytics.com/hub/cloud-training."
                 )
                 shutil.copy(last, weights)  # copy last.pt to best.pt
             else:

ultralytics/models/fastsam/model.py

@@ -16,8 +16,8 @@ class FastSAM(Model):
         ```python
         from ultralytics import FastSAM
 
-        model = FastSAM('last.pt')
-        results = model.predict('ultralytics/assets/bus.jpg')
+        model = FastSAM("last.pt")
+        results = model.predict("ultralytics/assets/bus.jpg")
         ```
     """
 
@@ -30,18 +30,21 @@ class FastSAM(Model):
 
     def predict(self, source, stream=False, bboxes=None, points=None, labels=None, texts=None, **kwargs):
         """
-        Performs segmentation prediction on the given image or video source.
+        Perform segmentation prediction on image or video source.
+
+        Supports prompted segmentation with bounding boxes, points, labels, and texts.
 
         Args:
-            source (str): Path to the image or video file, or a PIL.Image object, or a numpy.ndarray object.
-            stream (bool, optional): If True, enables real-time streaming. Defaults to False.
-            bboxes (list, optional): List of bounding box coordinates for prompted segmentation. Defaults to None.
-            points (list, optional): List of points for prompted segmentation. Defaults to None.
-            labels (list, optional): List of labels for prompted segmentation. Defaults to None.
-            texts (list, optional): List of texts for prompted segmentation. Defaults to None.
+            source (str | PIL.Image | numpy.ndarray): Input source.
+            stream (bool): Enable real-time streaming.
+            bboxes (list): Bounding box coordinates for prompted segmentation.
+            points (list): Points for prompted segmentation.
+            labels (list): Labels for prompted segmentation.
+            texts (list): Texts for prompted segmentation.
+            **kwargs (Any): Additional keyword arguments.
 
         Returns:
-            (list): The model predictions.
+            (list): Model predictions.
         """
         prompts = dict(bboxes=bboxes, points=points, labels=labels, texts=texts)
         return super().predict(source, stream, prompts=prompts, **kwargs)

ultralytics/models/fastsam/predict.py

@@ -92,8 +92,8 @@ class FastSAMPredictor(SegmentationPredictor):
                     if labels.sum() == 0  # all negative points
                     else torch.zeros(len(result), dtype=torch.bool, device=self.device)
                 )
-                for p, l in zip(points, labels):
-                    point_idx[torch.nonzero(masks[:, p[1], p[0]], as_tuple=True)[0]] = True if l else False
+                for point, label in zip(points, labels):
+                    point_idx[torch.nonzero(masks[:, point[1], point[0]], as_tuple=True)[0]] = True if label else False
                 idx |= point_idx
             if texts is not None:
                 if isinstance(texts, str):

ultralytics/models/fastsam/utils.py

@@ -13,7 +13,6 @@ def adjust_bboxes_to_image_border(boxes, image_shape, threshold=20):
     Returns:
         adjusted_boxes (torch.Tensor): adjusted bounding boxes
     """
-
     # Image dimensions
     h, w = image_shape
 

ultralytics/models/nas/model.py

@@ -6,8 +6,8 @@ Example:
     ```python
     from ultralytics import NAS
 
-    model = NAS('yolo_nas_s')
-    results = model.predict('ultralytics/assets/bus.jpg')
+    model = NAS("yolo_nas_s")
+    results = model.predict("ultralytics/assets/bus.jpg")
     ```
 """
 
@@ -34,8 +34,8 @@ class NAS(Model):
         ```python
         from ultralytics import NAS
 
-        model = NAS('yolo_nas_s')
-        results = model.predict('ultralytics/assets/bus.jpg')
+        model = NAS("yolo_nas_s")
+        results = model.predict("ultralytics/assets/bus.jpg")
         ```
 
     Attributes:

ultralytics/models/nas/predict.py

@@ -22,7 +22,7 @@ class NASPredictor(BasePredictor):
         ```python
         from ultralytics import NAS
 
-        model = NAS('yolo_nas_s')
+        model = NAS("yolo_nas_s")
         predictor = model.predictor
         # Assumes that raw_preds, img, orig_imgs are available
         results = predictor.postprocess(raw_preds, img, orig_imgs)
@@ -34,7 +34,6 @@ class NASPredictor(BasePredictor):
 
     def postprocess(self, preds_in, img, orig_imgs):
         """Postprocess predictions and returns a list of Results objects."""
-
         # Cat boxes and class scores
         boxes = ops.xyxy2xywh(preds_in[0][0])
         preds = torch.cat((boxes, preds_in[0][1]), -1).permute(0, 2, 1)

ultralytics/models/nas/val.py

@@ -24,7 +24,7 @@ class NASValidator(DetectionValidator):
         ```python
         from ultralytics import NAS
 
-        model = NAS('yolo_nas_s')
+        model = NAS("yolo_nas_s")
         validator = model.validator
         # Assumes that raw_preds are available
         final_preds = validator.postprocess(raw_preds)

ultralytics/models/rtdetr/predict.py

@@ -21,7 +21,7 @@ class RTDETRPredictor(BasePredictor):
         from ultralytics.utils import ASSETS
         from ultralytics.models.rtdetr import RTDETRPredictor
 
-        args = dict(model='rtdetr-l.pt', source=ASSETS)
+        args = dict(model="rtdetr-l.pt", source=ASSETS)
         predictor = RTDETRPredictor(overrides=args)
         predictor.predict_cli()
         ```

ultralytics/models/rtdetr/train.py

@@ -25,7 +25,7 @@ class RTDETRTrainer(DetectionTrainer):
         ```python
         from ultralytics.models.rtdetr.train import RTDETRTrainer
 
-        args = dict(model='rtdetr-l.yaml', data='coco8.yaml', imgsz=640, epochs=3)
+        args = dict(model="rtdetr-l.yaml", data="coco8.yaml", imgsz=640, epochs=3)
         trainer = RTDETRTrainer(overrides=args)
         trainer.train()
         ```

ultralytics/models/rtdetr/val.py

@@ -62,7 +62,7 @@ class RTDETRValidator(DetectionValidator):
         ```python
         from ultralytics.models.rtdetr import RTDETRValidator
 
-        args = dict(model='rtdetr-l.pt', data='coco8.yaml')
+        args = dict(model="rtdetr-l.pt", data="coco8.yaml")
         validator = RTDETRValidator(args=args)
         validator()
         ```