Prepro 210x160x3 uint8 frame into 6400 (80x80) 1D float vector.
(I)
| 45 | |
| 46 | |
| 47 | def prepro(I): |
| 48 | """Prepro 210x160x3 uint8 frame into 6400 (80x80) 1D float vector.""" |
| 49 | I = I[35:195] |
| 50 | I = I[::2, ::2, 0] |
| 51 | I[I == 144] = 0 |
| 52 | I[I == 109] = 0 |
| 53 | I[I != 0] = 1 |
| 54 | return I.astype(np.float32).ravel() |
| 55 | |
| 56 | |
| 57 | env = gym.make("Pong-v0") |
no outgoing calls
no test coverage detected
searching dependent graphs…