diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 68514863dcc..f5697b859c5 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -32,7 +32,7 @@ repos:
name: Upgrade code
- repo: https://github.com/PyCQA/isort
- rev: 5.12.0
+ rev: 5.13.2
hooks:
- id: isort
name: Sort imports
diff --git a/docs/en/integrations/index.md b/docs/en/integrations/index.md
index ca66549cf11..e7a917c3bdb 100644
--- a/docs/en/integrations/index.md
+++ b/docs/en/integrations/index.md
@@ -30,13 +30,15 @@ Welcome to the Ultralytics Integrations page! This page provides an overview of
- [Ray Tune](ray-tune.md): Optimize the hyperparameters of your Ultralytics models at any scale.
-- [TensorBoard](https://tensorboard.dev/): Visualize your Ultralytics ML workflows, monitor model metrics, and foster team collaboration.
+- [TensorBoard](tensorboard.md): Visualize your Ultralytics ML workflows, monitor model metrics, and foster team collaboration.
- [Weights & Biases (W&B)](weights-biases.md): Monitor experiments, visualize metrics, and foster reproducibility and collaboration on Ultralytics projects.
+- [Amazon SageMaker](amazon-sagemaker.md): Leverage Amazon SageMaker to efficiently build, train, and deploy Ultralytics models, providing an all-in-one platform for the ML lifecycle.
+
## Deployment Integrations
-- [Neural Magic](https://neuralmagic.com/): Leverage Quantization Aware Training (QAT) and pruning techniques to optimize Ultralytics models for superior performance and leaner size.
+- [Neural Magic](neural-magic.md): Leverage Quantization Aware Training (QAT) and pruning techniques to optimize Ultralytics models for superior performance and leaner size.
- [OpenVino](openvino.md): OpenVINO is Intel's toolkit for optimizing and deploying computer vision models efficiently across various Intel hardware platforms.
diff --git a/docs/en/reference/data/augment.md b/docs/en/reference/data/augment.md
index 3b16090c1f8..1d4099fca1a 100644
--- a/docs/en/reference/data/augment.md
+++ b/docs/en/reference/data/augment.md
@@ -79,10 +79,6 @@ keywords: Ultralytics, Data Augmentation, BaseTransform, MixUp, RandomHSV, Lette
-## ::: ultralytics.data.augment.hsv2colorjitter
-
-
-
-## ::: ultralytics.data.augment.classify_albumentations
+## ::: ultralytics.data.augment.classify_augmentations
diff --git a/docs/mkdocs.yml b/docs/mkdocs.yml
index c7ce51c0934..61b060033b6 100644
--- a/docs/mkdocs.yml
+++ b/docs/mkdocs.yml
@@ -289,6 +289,9 @@ nav:
- ClearML: integrations/clearml.md
- DVC: integrations/dvc.md
- Weights & Biases: integrations/weights-biases.md
+ - Neural Magic: integrations/neural-magic.md
+ - TensorBoard: integrations/tensorboard.md
+ - Amazon SageMaker: integrations/amazon-sagemaker.md
- Usage:
- CLI: usage/cli.md
- Python: usage/python.md
@@ -416,8 +419,8 @@ nav:
- tasks: reference/nn/tasks.md
- solutions:
- ai_gym: reference/solutions/ai_gym.md
- - object_counter: reference/solutions/object_counter.md
- heatmap: reference/solutions/heatmap.md
+ - object_counter: reference/solutions/object_counter.md
- trackers:
- basetrack: reference/trackers/basetrack.md
- bot_sort: reference/trackers/bot_sort.md
diff --git a/pyproject.toml b/pyproject.toml
index 7d6ae94e58b..4229d1d685f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,5 +1,5 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license
-#
+
# Overview:
# This pyproject.toml file manages the build, packaging, and distribution of the Ultralytics library.
# It defines essential project metadata, dependencies, and settings used to develop and deploy the library.
@@ -159,6 +159,9 @@ space_between_ending_comma_and_closing_bracket = true
split_before_closing_bracket = false
split_before_first_argument = false
+[tool.ruff]
+line-length = 120
+
[tool.docformatter]
wrap-summaries = 120
wrap-descriptions = 120
@@ -167,5 +170,5 @@ pre-summary-newline = true
close-quotes-on-newline = true
[tool.codespell]
-ignore-words-list = "crate,nd,strack,dota,ane,segway,fo,gool,winn"
-skip = '*.csv,*venv*,docs/de,docs/fr,docs/pt,docs/es,docs/mkdocs_de.yml'
+ignore-words-list = "crate,nd,strack,dota,ane,segway,fo,gool,winn,commend"
+skip = '*.csv,*venv*,docs/??/,docs/mkdocs_??.yml'
diff --git a/ultralytics/data/augment.py b/ultralytics/data/augment.py
index 97eb5fdba65..73728d7346d 100644
--- a/ultralytics/data/augment.py
+++ b/ultralytics/data/augment.py
@@ -1005,7 +1005,7 @@ def classify_transforms(
crop_fraction (float): fraction of image to crop. default is 1.0.
Returns:
- T.Compose: torchvision transforms
+ (T.Compose): torchvision transforms
"""
if isinstance(size, (tuple, list)):
@@ -1064,13 +1064,12 @@ def classify_augmentations(
hsv_h (float): image HSV-Hue augmentation (fraction)
hsv_s (float): image HSV-Saturation augmentation (fraction)
hsv_v (float): image HSV-Value augmentation (fraction)
- contrast (float): image contrast augmentation (fraction)
force_color_jitter (bool): force to apply color jitter even if auto augment is enabled
erasing (float): probability of random erasing
interpolation (T.InterpolationMode): interpolation mode. default is T.InterpolationMode.BILINEAR.
Returns:
- T.Compose: torchvision transforms
+ (T.Compose): torchvision transforms
"""
# Transforms to apply if albumentations not installed
if not isinstance(size, int):