(self, label_json_list, ratio_sample)
| 422 | ] |
| 423 | |
| 424 | def load_label_json(self, label_json_list, ratio_sample): |
| 425 | img_label_pair_list = {} |
| 426 | for line_json, rti_sam in zip(label_json_list, ratio_sample): |
| 427 | with open(line_json, 'r') as f: |
| 428 | json_data_list = json.load(f) |
| 429 | for keywh, value_list in json_data_list.items(): |
| 430 | if rti_sam > 1: |
| 431 | value_list = value_list * rti_sam |
| 432 | |
| 433 | w_r, h_r = keywh.split('_') |
| 434 | w_r = int(w_r) |
| 435 | h_r = int(h_r) |
| 436 | w_r, h_r = resize_image(w_r, h_r, self.max_side[0], |
| 437 | self.max_side[1]) |
| 438 | h_r = max( |
| 439 | int(h_r // self.divided_factor[1] * |
| 440 | self.divided_factor[1]), self.divided_factor[1]) |
| 441 | w_r = max( |
| 442 | int(w_r // self.divided_factor[0] * |
| 443 | self.divided_factor[0]), self.divided_factor[0]) |
| 444 | |
| 445 | key = str(w_r) + '_' + str(h_r) |
| 446 | if key in img_label_pair_list: |
| 447 | img_label_pair_list[key].extend(value_list) |
| 448 | else: |
| 449 | img_label_pair_list[key] = value_list |
| 450 | return img_label_pair_list |
| 451 | |
| 452 | def __init_lmdb(self): |
| 453 | """Initializes the LMDB environment.""" |
no test coverage detected