diff --git a/docker/Dockerfile b/docker/Dockerfile
index a561a72..4fca588 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -4,6 +4,7 @@
# Start FROM PyTorch image https://hub.docker.com/r/pytorch/pytorch or nvcr.io/nvidia/pytorch:23.03-py3
FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime
+RUN pip install --no-cache nvidia-tensorrt --index-url https://pypi.ngc.nvidia.com
# Downloads to user config dir
ADD https://ultralytics.com/assets/Arial.ttf https://ultralytics.com/assets/Arial.Unicode.ttf /root/.config/Ultralytics/
@@ -29,7 +30,7 @@ ADD https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt /u
# Install pip packages
RUN python3 -m pip install --upgrade pip wheel
-RUN pip install --no-cache -e . albumentations comet tensorboard thop pycocotools
+RUN pip install --no-cache -e . albumentations comet thop pycocotools onnx onnx-simplifier onnxruntime-gpu
# Set environment variables
ENV OMP_NUM_THREADS=1
diff --git a/docs/reference/nn/modules/conv.md b/docs/reference/nn/modules/conv.md
index 70591f6..7cfaf01 100644
--- a/docs/reference/nn/modules/conv.md
+++ b/docs/reference/nn/modules/conv.md
@@ -7,6 +7,11 @@ description: Explore convolutional neural network modules & techniques such as L
:::ultralytics.nn.modules.conv.Conv
+# Conv2
+---
+:::ultralytics.nn.modules.conv.Conv2
+
+
# LightConv
---
:::ultralytics.nn.modules.conv.LightConv
diff --git a/docs/reference/yolo/data/augment.md b/docs/reference/yolo/data/augment.md
index ed6c063..1cb38d0 100644
--- a/docs/reference/yolo/data/augment.md
+++ b/docs/reference/yolo/data/augment.md
@@ -87,6 +87,11 @@ description: Use Ultralytics YOLO Data Augmentation transforms with Base, MixUp,
:::ultralytics.yolo.data.augment.classify_transforms
+# hsv2colorjitter
+---
+:::ultralytics.yolo.data.augment.hsv2colorjitter
+
+
# classify_albumentations
---
:::ultralytics.yolo.data.augment.classify_albumentations
diff --git a/docs/reference/yolo/utils/benchmarks.md b/docs/reference/yolo/utils/benchmarks.md
index 4febb11..e3abcad 100644
--- a/docs/reference/yolo/utils/benchmarks.md
+++ b/docs/reference/yolo/utils/benchmarks.md
@@ -2,6 +2,11 @@
description: Improve your YOLO's performance and measure its speed. Benchmark utility for YOLOv5.
---
+# ProfileModels
+---
+:::ultralytics.yolo.utils.benchmarks.ProfileModels
+
+
# benchmark
---
:::ultralytics.yolo.utils.benchmarks.benchmark
diff --git a/docs/reference/yolo/utils/torch_utils.md b/docs/reference/yolo/utils/torch_utils.md
index 16ac301..f8fe445 100644
--- a/docs/reference/yolo/utils/torch_utils.md
+++ b/docs/reference/yolo/utils/torch_utils.md
@@ -57,11 +57,21 @@ description: Optimize your PyTorch models with Ultralytics YOLO's torch_utils fu
:::ultralytics.yolo.utils.torch_utils.get_num_gradients
+# model_info_for_loggers
+---
+:::ultralytics.yolo.utils.torch_utils.model_info_for_loggers
+
+
# get_flops
---
:::ultralytics.yolo.utils.torch_utils.get_flops
+# get_flops_with_torch_profiler
+---
+:::ultralytics.yolo.utils.torch_utils.get_flops_with_torch_profiler
+
+
# initialize_weights
---
:::ultralytics.yolo.utils.torch_utils.initialize_weights
diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py
index 5b3beda..ebca26d 100644
--- a/ultralytics/__init__.py
+++ b/ultralytics/__init__.py
@@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license
-__version__ = '8.0.109'
+__version__ = '8.0.110'
from ultralytics.hub import start
from ultralytics.vit.rtdetr import RTDETR
diff --git a/ultralytics/yolo/data/augment.py b/ultralytics/yolo/data/augment.py
index f8476ba..4c7f167 100644
--- a/ultralytics/yolo/data/augment.py
+++ b/ultralytics/yolo/data/augment.py
@@ -778,11 +778,8 @@ def v8_transforms(dataset, imgsz, hyp):
if flip_idx is None and hyp.fliplr > 0.0:
hyp.fliplr = 0.0
LOGGER.warning("WARNING ⚠️ No 'flip_idx' array defined in data.yaml, setting augmentation 'fliplr=0.0'")
- elif flip_idx:
- if len(flip_idx) != kpt_shape[0]:
- raise ValueError(f'data.yaml flip_idx={flip_idx} length must be equal to kpt_shape[0]={kpt_shape[0]}')
- elif flip_idx[0] != 0:
- raise ValueError(f'data.yaml flip_idx={flip_idx} must be zero-index (start from 0)')
+ elif flip_idx and (len(flip_idx) != kpt_shape[0]):
+ raise ValueError(f'data.yaml flip_idx={flip_idx} length must be equal to kpt_shape[0]={kpt_shape[0]}')
return Compose([
pre_transform,