(grid_z, grid_y, grid_x, Z, Y, X, clamp_extreme=True)
| 181 | return xy |
| 182 | |
| 183 | def normalize_grid3d(grid_z, grid_y, grid_x, Z, Y, X, clamp_extreme=True): |
| 184 | # make things in [-1,1] |
| 185 | grid_z = 2.0*(grid_z / float(Z-1)) - 1.0 |
| 186 | grid_y = 2.0*(grid_y / float(Y-1)) - 1.0 |
| 187 | grid_x = 2.0*(grid_x / float(X-1)) - 1.0 |
| 188 | |
| 189 | if clamp_extreme: |
| 190 | grid_z = torch.clamp(grid_z, min=-2.0, max=2.0) |
| 191 | grid_y = torch.clamp(grid_y, min=-2.0, max=2.0) |
| 192 | grid_x = torch.clamp(grid_x, min=-2.0, max=2.0) |
| 193 | |
| 194 | return grid_z, grid_y, grid_x |
| 195 | |
| 196 | def normalize_grid2d(grid_y, grid_x, Y, X, clamp_extreme=True): |
| 197 | # make things in [-1,1] |
no outgoing calls
no test coverage detected