(self,phase='train',image_size=256)
| 130 | |
| 131 | class SBI_API: |
| 132 | def __init__(self,phase='train',image_size=256): |
| 133 | |
| 134 | assert phase == 'train', f"Current SBI API only support train phase, but got {phase}" |
| 135 | |
| 136 | self.image_size=(image_size,image_size) |
| 137 | self.phase=phase |
| 138 | |
| 139 | self.transforms=self.get_transforms() |
| 140 | self.source_transforms = self.get_source_transforms() |
| 141 | self.bob_transforms = self.get_source_transforms_for_bob() |
| 142 | |
| 143 | |
| 144 | def __call__(self,img,landmark=None,sri_image=None): |
nothing calls this directly
no test coverage detected