Skip to content

Latest commit

 

History

History
46 lines (34 loc) · 1.61 KB

compatibility.md

File metadata and controls

46 lines (34 loc) · 1.61 KB

Compatibility of MMYOLO

MMYOLO 0.3.0

METAINFO modification

To unify with other OpenMMLab repositories, change all keys of METAINFO in Dataset from upper case to lower case.

Before v0.3.0 after v0.3.0
CLASSES classes
PALETTE palette
DATASET_TYPE dataset_type

About the order of image shape

In OpenMMLab 2.0, to be consistent with the input argument of OpenCV, the argument about image shape in the data transformation pipeline is always in the (width, height) order. On the contrary, for computation convenience, the order of the field going through the data pipeline and the model is (height, width). Specifically, in the results processed by each data transform pipeline, the fields and their value meaning is as below:

  • img_shape: (height, width)
  • ori_shape: (height, width)
  • pad_shape: (height, width)
  • batch_input_shape: (height, width)

As an example, the initialization arguments of Mosaic are as below:

@TRANSFORMS.register_module()
class Mosaic(BaseTransform):
    def __init__(self,
                img_scale: Tuple[int, int] = (640, 640),
                center_ratio_range: Tuple[float, float] = (0.5, 1.5),
                bbox_clip_border: bool = True,
                pad_val: float = 114.0,
                prob: float = 1.0) -> None:
       ...

       # img_scale order should be (width, height)
       self.img_scale = img_scale

    def transform(self, results: dict) -> dict:
        ...

        results['img'] = mosaic_img
        # (height, width)
        results['img_shape'] = mosaic_img.shape[:2]