(self, img_dir=None, out_path=None)
| 16 | |
| 17 | class INFERENCE_BEDLAM(torch.utils.data.Dataset): |
| 18 | def __init__(self, img_dir=None, out_path=None): |
| 19 | self.img_dir = os.path.join(img_dir, '**/*.png') |
| 20 | self.out_path = out_path |
| 21 | self.img_paths = sorted(glob(self.img_dir,recursive=True)) |
| 22 | |
| 23 | self.score_threshold = cfg.threshold if 'threshold' in cfg else 0.85 |
| 24 | self.resolution = [720,1280] |
| 25 | self.format = DefaultFormatBundle() |
| 26 | self.normalize = Normalize(mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375]) |
| 27 | |
| 28 | def __len__(self): |
| 29 | return len(self.img_paths) |
nothing calls this directly
no test coverage detected