(self, origin_img)
| 52 | return len(self.raw_data) |
| 53 | |
| 54 | def __resize__(self, origin_img): |
| 55 | resolution = origin_img.size |
| 56 | w,h = resolution |
| 57 | if self.max_line_res is not None: |
| 58 | max_line = self.max_line_res |
| 59 | if h > max_line: |
| 60 | w = int(w * max_line / h) |
| 61 | h = max_line |
| 62 | if w > max_line: |
| 63 | h = int(h * max_line / w) |
| 64 | w = max_line |
| 65 | img = origin_img.resize((w,h),resample=Image.Resampling.LANCZOS) |
| 66 | return img |
| 67 | |
| 68 | def __getitem__(self, i) -> Dict[str, torch.Tensor]: |
| 69 | try: |