(batch, ht, wd)
| 41 | print('video saved') |
| 42 | |
| 43 | def coords_grid(batch, ht, wd): |
| 44 | coords = torch.meshgrid(torch.arange(ht), torch.arange(wd)) |
| 45 | coords = torch.stack(coords[::-1], dim=0).float() |
| 46 | |
| 47 | return coords[None].repeat(batch, 1, 1, 1) |
| 48 | |
| 49 | def image_flow_warp(image, flow, padding_mode='zeros'): |
| 50 | ''' |
no outgoing calls
no test coverage detected