(im, size)
| 33 | |
| 34 | |
| 35 | def resize_keepdims(im, size): |
| 36 | # Opencv's resize remove the extra dimension for grayscale images. We add it back. |
| 37 | ret = cv2.resize(im, size) |
| 38 | if im.ndim == 3 and ret.ndim == 2: |
| 39 | ret = ret[:, :, np.newaxis] |
| 40 | return ret |
| 41 | |
| 42 | |
| 43 | def get_player(viz=False, train=False): |