From 7a48b62cbc4a2fa709ee4bb86e8c95a693c08e1f Mon Sep 17 00:00:00 2001 From: Burhan <62214284+Burhan-Q@users.noreply.github.com> Date: Thu, 25 Jul 2024 11:02:19 -0400 Subject: [PATCH] Remove duplicate `make_divisible` function (#14690) Co-authored-by: UltralyticsAssistant --- docs/en/reference/utils/torch_utils.md | 4 ---- ultralytics/nn/tasks.py | 2 +- ultralytics/utils/torch_utils.py | 7 ------- 3 files changed, 1 insertion(+), 12 deletions(-) diff --git a/docs/en/reference/utils/torch_utils.md b/docs/en/reference/utils/torch_utils.md index dd4c364d..1a196831 100644 --- a/docs/en/reference/utils/torch_utils.md +++ b/docs/en/reference/utils/torch_utils.md @@ -83,10 +83,6 @@ keywords: Ultralytics, torch utils, model optimization, device selection, infere



-## ::: ultralytics.utils.torch_utils.make_divisible - -



- ## ::: ultralytics.utils.torch_utils.copy_attr



diff --git a/ultralytics/nn/tasks.py b/ultralytics/nn/tasks.py index a30094c9..f6feed23 100644 --- a/ultralytics/nn/tasks.py +++ b/ultralytics/nn/tasks.py @@ -66,13 +66,13 @@ from ultralytics.utils.loss import ( v8PoseLoss, v8SegmentationLoss, ) +from ultralytics.utils.ops import make_divisible from ultralytics.utils.plotting import feature_visualization from ultralytics.utils.torch_utils import ( fuse_conv_and_bn, fuse_deconv_and_bn, initialize_weights, intersect_dicts, - make_divisible, model_info, scale_img, time_sync, diff --git a/ultralytics/utils/torch_utils.py b/ultralytics/utils/torch_utils.py index fcecd148..d2204303 100644 --- a/ultralytics/utils/torch_utils.py +++ b/ultralytics/utils/torch_utils.py @@ -424,13 +424,6 @@ def scale_img(img, ratio=1.0, same_shape=False, gs=32): return F.pad(img, [0, w - s[1], 0, h - s[0]], value=0.447) # value = imagenet mean -def make_divisible(x, divisor): - """Returns nearest x divisible by divisor.""" - if isinstance(divisor, torch.Tensor): - divisor = int(divisor.max()) # to int - return math.ceil(x / divisor) * divisor - - def copy_attr(a, b, include=(), exclude=()): """Copies attributes from object 'b' to object 'a', with options to include/exclude certain attributes.""" for k, v in b.__dict__.items():