Spaces:
Runtime error
Runtime error
| #################### | |
| # Tasks | |
| #################### | |
| task_parameters = { | |
| 'class_object':{ | |
| 'num_classes': 1000, | |
| 'ext': 'npy', | |
| 'domain_id': 'class_object', | |
| }, | |
| 'class_scene':{ | |
| 'num_classes': 365, | |
| 'ext': 'npy', | |
| 'domain_id': 'class_scene', | |
| }, | |
| 'depth_zbuffer':{ | |
| 'num_channels': 1, | |
| 'mask_val': 1.0, | |
| 'clamp_to': (0.0, 8000.0 / (2**16 - 1)), # Same as consistency | |
| 'ext': 'png', | |
| 'domain_id': 'depth_zbuffer', | |
| }, | |
| 'depth_euclidean':{ | |
| 'num_channels': 1, | |
| 'clamp_to': (0.0, 8000.0 / (2**16 - 1)), # Same as consistency | |
| # 'mask_val': 1.0, | |
| 'ext': 'png', | |
| 'domain_id': 'depth_euclidean', | |
| }, | |
| 'edge_texture': { | |
| 'num_channels': 1, | |
| 'clamp_to': (0.0, 0.25), | |
| #'threshold_min': 0.01, | |
| 'ext': 'png', | |
| 'domain_id': 'edge_texture', | |
| }, | |
| 'edge_occlusion': { | |
| 'num_channels': 1, | |
| #'clamp_to': (0.0, 0.04), | |
| #'threshold_min': 0.0017, | |
| 'ext': 'png', | |
| 'domain_id': 'edge_occlusion', | |
| }, | |
| 'keypoints3d': { | |
| 'num_channels': 1, | |
| 'ext': 'png', | |
| 'domain_id': 'keypoints3d', | |
| }, | |
| 'keypoints2d':{ | |
| 'num_channels': 1, | |
| #'clamp_to': (0.0, 0.025), | |
| #'threshold_min': 0.002, | |
| 'ext': 'png', | |
| 'domain_id': 'keypoints2d', | |
| }, | |
| 'principal_curvature':{ | |
| 'num_channels': 3, | |
| 'mask_val': 0.0, | |
| 'ext': 'png', | |
| 'domain_id': 'principal_curvature', | |
| }, | |
| 'reshading':{ | |
| 'num_channels': 1, | |
| 'ext': 'png', | |
| 'domain_id': 'reshading', | |
| }, | |
| 'normal':{ | |
| 'num_channels': 3, | |
| 'mask_val': 0.502, | |
| 'ext': 'png', | |
| 'domain_id': 'normal', | |
| }, | |
| 'mask_valid':{ | |
| 'num_channels': 1, | |
| 'mask_val': 0.0, | |
| 'ext': 'png', | |
| 'domain_id': 'depth_zbuffer', | |
| }, | |
| 'rgb':{ | |
| 'num_channels': 3, | |
| 'ext': 'png', | |
| 'domain_id': 'rgb', | |
| }, | |
| 'segment_semantic': { | |
| 'num_channels': 18, | |
| 'ext': 'png', | |
| 'domain_id': 'segmentsemantic', | |
| }, | |
| 'segment_unsup2d':{ | |
| 'num_channels': 64, | |
| 'ext': 'png', | |
| 'domain_id': 'segment_unsup2d', | |
| }, | |
| 'segment_unsup25d':{ | |
| 'num_channels': 64, | |
| 'ext': 'png', | |
| 'domain_id': 'segment_unsup25d', | |
| }, | |
| } | |
| PIX_TO_PIX_TASKS = ['colorization', 'edge_texture', 'edge_occlusion', 'keypoints3d', 'keypoints2d', 'reshading', 'depth_zbuffer', 'depth_euclidean', 'curvature', 'autoencoding', 'denoising', 'normal', 'inpainting', 'segment_unsup2d', 'segment_unsup25d', 'segment_semantic', ] | |
| FEED_FORWARD_TASKS = ['class_object', 'class_scene', 'room_layout', 'vanishing_point'] | |
| SINGLE_IMAGE_TASKS = PIX_TO_PIX_TASKS + FEED_FORWARD_TASKS | |
| SIAMESE_TASKS = ['fix_pose', 'jigsaw', 'ego_motion', 'point_match', 'non_fixated_pose'] | |