| 22 | |
| 23 | |
| 24 | def get_pixel_range_prob_dict(pixel_num_list: list, pixel_prob_list: list): |
| 25 | if len(pixel_num_list) == len(pixel_prob_list): |
| 26 | pixel_num_list.append(pixel_num_list[-1] * 4) |
| 27 | assert len(pixel_prob_list) == len(pixel_num_list) - 1 |
| 28 | pixel_prob_list = np.array(pixel_prob_list) |
| 29 | pixel_prob_list = pixel_prob_list / pixel_prob_list.sum() |
| 30 | prob_dict = OrderedDict({}) |
| 31 | for i in range(len(pixel_prob_list)): |
| 32 | pixel_range = (pixel_num_list[i], pixel_num_list[i + 1]) |
| 33 | prob_dict[pixel_range] = pixel_prob_list[i] |
| 34 | return prob_dict |
| 35 | |
| 36 | |
| 37 | def get_bucket_config_from_side_list_and_ar_list( |