(grid)
| 235 | return reverse_mask |
| 236 | |
| 237 | def refine_grid(grid): |
| 238 | H, W, _ = grid.shape |
| 239 | grid[:,:,0] = (grid[:,:,0] + 1/2) * W |
| 240 | grid[:,:,1] = (grid[:,:,1] + 1/2) * H |
| 241 | |
| 242 | return grid |
| 243 | |
| 244 | def convert_flow_to_mapping(flow, output_channel_first=True): |
| 245 | if not isinstance(flow, np.ndarray): |