Class BucketConfigHardCoded1

datasets/bucket_config.py:267–280

class BucketConfigHardCoded1(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([[480, 720, 480 * 720]],
                             dtype=np.int64), np.array([1.0])),
            1.0: (np.array([[384, 384, 384**2], [512, 512, 512**2]],
                           dtype=np.int64), np.array([0.4, 0.6])),
            1.5: (np.array([[720, 480, 480 * 720]],
                           dtype=np.int64), np.array([1.0])),
        }
        super().__init__()
        pass


class BucketConfigHardCoded2(BucketConfig):