29 lines
1.2 KiB
Python
29 lines
1.2 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
def make_divisible(value, divisor, min_value=None, min_ratio=0.9):
|
|
"""Make divisible function.
|
|
|
|
This function rounds the channel number to the nearest value that can be
|
|
divisible by the divisor. It is taken from the original tf repo. It ensures
|
|
that all layers have a channel number that is divisible by divisor. It can
|
|
be seen here: https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobilenet.py # noqa
|
|
|
|
Args:
|
|
value (int): The original channel number.
|
|
divisor (int): The divisor to fully divide the channel number.
|
|
min_value (int): The minimum value of the output channel.
|
|
Default: None, means that the minimum value equal to the divisor.
|
|
min_ratio (float): The minimum ratio of the rounded channel number to
|
|
the original channel number. Default: 0.9.
|
|
|
|
Returns:
|
|
int: The modified output channel number.
|
|
"""
|
|
|
|
if min_value is None:
|
|
min_value = divisor
|
|
new_value = max(min_value, int(value + divisor / 2) // divisor * divisor)
|
|
# Make sure that round down does not go down by more than (1-min_ratio).
|
|
if new_value < min_ratio * value:
|
|
new_value += divisor
|
|
return new_value
|