Remove duplicate make_divisible function (#14690)

Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Burhan 2024-07-25 11:02:19 -04:00 committed by GitHub
parent 2b95f9aec0
commit 7a48b62cbc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 1 additions and 12 deletions

View File

@ -83,10 +83,6 @@ keywords: Ultralytics, torch utils, model optimization, device selection, infere
<br><br><hr><br>
## ::: ultralytics.utils.torch_utils.make_divisible
<br><br><hr><br>
## ::: ultralytics.utils.torch_utils.copy_attr
<br><br><hr><br>

View File

@ -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,

View File

@ -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():