(
self,
file_list: str,
file_backend_cfg: Mapping[str, Any],
out_size: int,
crop_type: str,
blur_kernel_size: int,
kernel_list: Sequence[str],
kernel_prob: Sequence[float],
blur_sigma: Sequence[float],
downsample_range: Sequence[float],
noise_range: Sequence[float],
jpeg_range: Sequence[int],
)
| 21 | class CodeformerDataset(data.Dataset): |
| 22 | |
| 23 | def __init__( |
| 24 | self, |
| 25 | file_list: str, |
| 26 | file_backend_cfg: Mapping[str, Any], |
| 27 | out_size: int, |
| 28 | crop_type: str, |
| 29 | blur_kernel_size: int, |
| 30 | kernel_list: Sequence[str], |
| 31 | kernel_prob: Sequence[float], |
| 32 | blur_sigma: Sequence[float], |
| 33 | downsample_range: Sequence[float], |
| 34 | noise_range: Sequence[float], |
| 35 | jpeg_range: Sequence[int], |
| 36 | ) -> "CodeformerDataset": |
| 37 | super(CodeformerDataset, self).__init__() |
| 38 | self.file_list = file_list |
| 39 | self.image_files = load_file_list(file_list) |
| 40 | self.file_backend = instantiate_from_config(file_backend_cfg) |
| 41 | self.out_size = out_size |
| 42 | self.crop_type = crop_type |
| 43 | assert self.crop_type in ["none", "center", "random"] |
| 44 | # degradation configurations |
| 45 | self.blur_kernel_size = blur_kernel_size |
| 46 | self.kernel_list = kernel_list |
| 47 | self.kernel_prob = kernel_prob |
| 48 | self.blur_sigma = blur_sigma |
| 49 | self.downsample_range = downsample_range |
| 50 | self.noise_range = noise_range |
| 51 | self.jpeg_range = jpeg_range |
| 52 | |
| 53 | def load_gt_image( |
| 54 | self, image_path: str, max_retry: int = 5 |
nothing calls this directly
no test coverage detected