| 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) |
| |
| if new_value < min_ratio * value: |
| new_value += divisor |
| return new_value |
|
|
