Chooses static padding if you have specified an image size, and dynamic padding otherwise. Static padding is necessary for ONNX exporting of models. Args: image_size (int or tuple): Size of the image. Returns: Conv2dDynamicSamePadding or Conv2dStaticSamePadding.
(image_size=None)
| 107 | |
| 108 | |
| 109 | def get_same_padding_conv2d(image_size=None): |
| 110 | """Chooses static padding if you have specified an image size, and dynamic padding otherwise. |
| 111 | Static padding is necessary for ONNX exporting of models. |
| 112 | Args: |
| 113 | image_size (int or tuple): Size of the image. |
| 114 | Returns: |
| 115 | Conv2dDynamicSamePadding or Conv2dStaticSamePadding. |
| 116 | """ |
| 117 | if image_size is None: |
| 118 | return Conv2dDynamicSamePadding |
| 119 | else: |
| 120 | return partial(Conv2dStaticSamePadding, image_size=image_size) |
| 121 | |
| 122 | |
| 123 | class Conv2dDynamicSamePadding(nn.Conv2d): |
no outgoing calls
no test coverage detected