MCPcopy Create free account
hub / github.com/MotrixLab/ViMoGen / BucketConfigFromSideList

Class BucketConfigFromSideList

datasets/bucket_config.py:323–336  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

321
322
323class BucketConfigFromSideList(BucketConfig):
324
325 def __init__(self, side_list: list, ar_list: list, ar_max: float,
326 ar_min: float, pixel_num_list: list,
327 pixel_prob_list: list) -> None:
328 assert len(pixel_num_list) == len(pixel_prob_list) or len(
329 pixel_num_list) == (
330 len(pixel_prob_list) + 1)
331 prob_dict = get_pixel_range_prob_dict(pixel_num_list, pixel_prob_list)
332 bucket_config = get_bucket_config_from_side_list_and_ar_list(
333 side_list, ar_list, ar_max=ar_max, ar_min=ar_min)
334 self.bucket_config = assign_probability(bucket_config, prob_dict)
335 super().__init__()
336 pass
337
338
339class DefaultBucketConfig(BucketConfigFromSideList):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected