| 267 | class BucketConfigHardCoded1(BucketConfig): |
| 268 | |
| 269 | def __init__(self, ) -> None: |
| 270 | self.bucket_config = { |
| 271 | # aspect_ratio: (height width pixel value with shape of (n, 3), probability with shape of (n, ) ) |
| 272 | 0.667: (np.array([[480, 720, 480 * 720]], |
| 273 | dtype=np.int64), np.array([1.0])), |
| 274 | 1.0: (np.array([[384, 384, 384**2], [512, 512, 512**2]], |
| 275 | dtype=np.int64), np.array([0.4, 0.6])), |
| 276 | 1.5: (np.array([[720, 480, 480 * 720]], |
| 277 | dtype=np.int64), np.array([1.0])), |
| 278 | } |
| 279 | super().__init__() |
| 280 | pass |
| 281 | |
| 282 | |
| 283 | class BucketConfigHardCoded2(BucketConfig): |