| 8 | |
| 9 | |
| 10 | class ToNumpy: |
| 11 | |
| 12 | def __call__(self, pil_img): |
| 13 | np_img = np.array(pil_img, dtype=np.uint8) |
| 14 | if np_img.ndim < 3: |
| 15 | np_img = np.expand_dims(np_img, axis=-1) |
| 16 | np_img = np.rollaxis(np_img, 2) # HWC to CHW |
| 17 | return np_img |
| 18 | |
| 19 | |
| 20 | class ToTensor: |
no outgoing calls
no test coverage detected