in c3dm/dataset/eval_zoo.py [0:0]
def eval_zoo(dataset_name, include_debug_vars=False):
if 'freicars_clickp_filtd' in dataset_name:
eval_script = eval_freicars
cache_vars = [
'masks', 'depth_dense',
'K_orig', 'image_path',
'orig_image_size',
'depth_path', 'mask_path',
'seq_name', 'R', 'T',
'embed_db_shape_camera_coord',
'cmr_faces',
'kp_loc_3d',
'shape_image_coord_cal',
'nrsfm_shape_image_coord'
]
eval_vars = [
'EVAL_depth_scl_perspective_med',
'EVAL_pcl_scl_perspective_med',
'EVAL_pcl_corr_scl_perspective_med',
'EVAL_depth_scl_orthographic_med',
'EVAL_pcl_scl_orthographic_med',
'EVAL_pcl_corr_scl_orthographic_med',
'EVAL_depth_scl_perspective',
'EVAL_pcl_scl_perspective',
'EVAL_pcl_corr_scl_perspective',
'EVAL_depth_scl_orthographic',
'EVAL_pcl_scl_orthographic',
'EVAL_pcl_corr_scl_orthographic',
'EVAL_sparse_pcl',
'EVAL_sparse_pcl_nrsfm',
]
elif dataset_name in ('celeba_ff',):
eval_script = eval_florence
cache_vars = [ 'masks', 'depth_dense',
'K_orig', 'image_path',
'orig_image_size',
'depth_path', 'mask_path',
'seq_name',
'images',
'embed_db_shape_camera_coord',
'shape_image_coord_cal_dense',
'cmr_faces',
'kp_loc',
'mesh_path',
'shape_image_coord_best_scale',
]
eval_vars = [ \
'EVAL_pcl_scl_recut_orthographic_flip_med',
'EVAL_pcl_scl_orthographic_flip_med',
'EVAL_pcl_orthographic_flip_med',
'EVAL_pcl_scl_recut_orthographic_med',
'EVAL_pcl_scl_orthographic_med',
'EVAL_pcl_orthographic_med',
'EVAL_pcl_scl_recut_orthographic_flip',
'EVAL_pcl_scl_orthographic_flip',
'EVAL_pcl_orthographic_flip',
'EVAL_pcl_scl_recut_orthographic',
'EVAL_pcl_scl_orthographic',
'EVAL_pcl_orthographic',
]
elif 'pascal3d' in dataset_name:
eval_script = eval_p3d
cache_vars = [ 'masks', 'depth_dense',
'image_path', 'R', 'T',
'orig_image_size',
'mask_path',
'images',
'embed_db_shape_camera_coord',
'shape_image_coord_cal_dense',
'cmr_faces',
'kp_loc',
'mesh_path' ]
eval_vars = [ \
'EVAL_pcl_scl_detkp',
'EVAL_pcl_corr_scl_detkp',
'EVAL_pcl_corr_scl_detkp_med',
]
else:
eval_script = eval_dummy
cache_vars = [ 'images', ]
eval_vars = [ 'EVAL_pcl_dist_scl', ]
return eval_script, cache_vars, eval_vars