Padding bug

[nnistelrooij]

mmyolo/mmyolo/datasets/transforms/transforms.py

Lines 292 to 302 in 8c4d9dc

	top_padding, _, left_padding, _ = results['pad_param']
	if int(left_padding) != 0:
	gt_masks = gt_masks.translate(
	out_shape=results['img_shape'][:2],
	offset=int(left_padding),
	direction='horizontal')
	if int(top_padding) != 0:
	gt_masks = gt_masks.translate(
	out_shape=results['img_shape'][:2],
	offset=int(top_padding),
	direction='vertical')

The bottom and right padding may be positive, while the top and left padding is zero, causing an incorrect translation of the polygons and an incorrect output shape.

The code below addresses the bug:

        top_padding, bottom_padding, left_padding, right_padding = results['pad_param']
        if int(left_padding) != 0:
            gt_masks = gt_masks.translate(
                out_shape=results['img_shape'][:2],
                offset=int(left_padding),
                direction='horizontal')
        elif int(right_padding) != 0:
            gt_masks = gt_masks.translate(
                out_shape=results['img_shape'][:2],
                offset=-int(right_padding),
                direction='horizontal')
            
        if int(top_padding) != 0:
            gt_masks = gt_masks.translate(
                out_shape=results['img_shape'][:2],
                offset=int(top_padding),
                direction='vertical')
        elif int(bottom_padding) != 0:
            gt_masks = gt_masks.translate(
                out_shape=results['img_shape'][:2],
                offset=-int(bottom_padding),
                direction='vertical')