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