for gt_id, gt in enumerate(scene_gt[im_id]): if gt['obj_id'] != obj_id: continue err = -1.0 R_gt = gt['cam_R_m2c'] t_gt = gt['cam_t_m2c'] if error_type == 'vsd': err = pose_error.vsd(R_est, t_est, R_gt, t_gt, model, depth_im, delta, tau, K) elif error_type == 'add': err = pose_error.add(R_est, t_est, R_gt, t_gt, model) elif error_type == 'adi': err = pose_error.adi(R_est, t_est, R_gt, t_gt, model) elif error_type == 'cou': err = pose_error.cou(R_est, t_est, R_gt, t_gt, model, dp['test_im_size'], K) elif error_type == 're': err = pose_error.re(R_est, R_gt) elif error_type == 'te': err = pose_error.te(t_est, t_gt) errs.append([im_id, obj_id, est_id, gt_id, err]) #print('Evaluation time: {}s'.format(time.time() - t)) print('Saving calculated errors...') errs_path = errs_mpath.format(result_path=result_path, scene_id=scene_id, err_desc=err_desc)
for gt_id, gt in enumerate(scene_gt[im_id]): if gt['obj_id'] != obj_id: continue e = -1.0 R_g = gt['cam_R_m2c'] t_g = gt['cam_t_m2c'] if error_type == 'vsd': e = pose_error.vsd(R_e, t_e, R_g, t_g, models[obj_id], depth_im, K, vsd_delta, vsd_tau, vsd_cost) elif error_type == 'add': e = pose_error.add(R_e, t_e, R_g, t_g, models[obj_id]) elif error_type == 'adi': e = pose_error.adi(R_e, t_e, R_g, t_g, models[obj_id]) elif error_type == 'cou': e = pose_error.cou(R_e, t_e, R_g, t_g, models[obj_id], dp['test_im_size'], K) elif error_type == 're': e = pose_error.re(R_e, R_g) elif error_type == 'te': e = pose_error.te(t_e, t_g) errs_gts[gt_id] = e errs.append({ 'im_id': im_id, 'obj_id': obj_id, 'est_id': est_id, 'score': est['score'],
for gt_id, gt in enumerate(scene_gt[im_id]): if gt['obj_id'] != obj_id: continue e = -1.0 R_g = gt['cam_R_m2c'] t_g = gt['cam_t_m2c'] if error_type == 'vsd': e = pose_error.vsd(R_e, t_e, R_g, t_g, models[obj_id], depth_im, K, vsd_delta, vsd_tau, vsd_cost) elif error_type == 'add': e = pose_error.add(R_e, t_e, R_g, t_g, models[obj_id]) elif error_type == 'adi': e = pose_error.adi(R_e, t_e, R_g, t_g, models[obj_id]) elif error_type == 'cou': e = pose_error.cou(R_e, t_e, R_g, t_g, models[obj_id], dp['test_im_size'], K) elif error_type == 're': e = pose_error.re(R_e, R_g) elif error_type == 'te': e = pose_error.te(t_e, t_g) errs_gts[gt_id] = e errs.append({ 'im_id': im_id, 'obj_id': obj_id, 'est_id': est_id, 'score': est['score'],