(self, file_list, ratio_list)
| 73 | return |
| 74 | |
| 75 | def get_image_info_list(self, file_list, ratio_list): |
| 76 | if isinstance(file_list, str): |
| 77 | file_list = [file_list] |
| 78 | data_lines = [] |
| 79 | for idx, file in enumerate(file_list): |
| 80 | with open(file, 'rb') as f: |
| 81 | lines = f.readlines() |
| 82 | if self.mode == 'train' or ratio_list[idx] < 1.0: |
| 83 | random.seed(self.seed) |
| 84 | lines = random.sample(lines, |
| 85 | round(len(lines) * ratio_list[idx])) |
| 86 | data_lines.extend(lines) |
| 87 | return data_lines |
| 88 | |
| 89 | def shuffle_data_random(self): |
| 90 | random.seed(self.seed) |