downscale translation pose to [-1:1] only
(poses, sc)
| 104 | return poses[:,:3,:].reshape(poses.shape[0],12) |
| 105 | |
| 106 | def downscale_pose(poses, sc): |
| 107 | ''' downscale translation pose to [-1:1] only ''' |
| 108 | target_pose = poses.reshape(poses.shape[0],3,4) |
| 109 | target_pose[:,:3,3] = target_pose[:,:3,3] * sc |
| 110 | return target_pose.reshape(poses.shape[0],12) |
| 111 | |
| 112 | class Cambridge2(data.Dataset): |
| 113 | def __init__(self, scene, data_path, train, transform=None, |
nothing calls this directly
no outgoing calls
no test coverage detected