tfm.vision.preprocess_ops.compute_padded_size

Compute the padded size given the desired size and the stride.

The padded size will be the smallest rectangle, such that each dimension is the smallest multiple of the stride which is larger than the desired dimension. For example, if desired_size = (100, 200) and stride = 32, the output padded_size = (128, 224).

desired_size a Tensor or int list/tuple of two elements representing [height, width] of the target output image size.
stride an integer, the stride of the backbone network.

padded_size a Tensor or int list/tuple of two elements representing [height, width] of the padded output image size.