Spaces:
Running
Running
File size: 2,905 Bytes
3b49518 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
####################
# 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']
|