Class BucketConfigHardCoded3

datasets/bucket_config.py:300–320

class BucketConfigHardCoded3(BucketConfig):

    def __init__(self, ) -> None:
        self.bucket_config = {
            # aspect_ratio: (height width pixel value with shape of (n, 3), probability with shape of (n, ) )
            0.667: (np.array([[240, 360], [360, 540], [480, 720], [720, 1080]],
                             dtype=np.int64), np.array([
                                 0.2,
                                 0.5,
                                 0.2,
                                 0.1,
                             ])),
            1.0: (np.array([[256, 256], [384, 384], [512, 512], [768, 768]],
                           dtype=np.int64), np.array([0.2, 0.5, 0.2, 0.1])),
            1.5:
            (np.array([[240, 360], [360, 540], [480, 720], [720, 1080]],
                      dtype=np.int64)[:, ::-1], np.array([0.2, 0.5, 0.2,
                                                          0.1])),
        }
        super().__init__()
        pass


class BucketConfigFromSideList(BucketConfig):