mirror of
https://github.com/ultralytics/ultralytics.git
synced 2025-09-15 15:48:41 +08:00
Remove duplicate make_divisible function (#14690)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
parent
2b95f9aec0
commit
7a48b62cbc
@ -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>
|
||||
|
||||
@ -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,
|
||||
|
||||
@ -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():
|
||||
|
||||
Loading…
Reference in New Issue
Block a user