Exemple #1
0
    depth_gt[(mask * (-1) + 1).astype(np.bool)] = 0
    depth_asw[(mask * (-1) + 1).astype(np.bool)] = 0

    rmse_aux_depth = rmse(depth_gt, depth_asw, mask)
    mae_aux_depth = mae(depth_gt, depth_asw, mask)

    global_rmse_depth += rmse_aux_depth
    global_mae_depth += mae_aux_depth

    global_rmse_disp += rmse_aux_disp
    global_mae_disp += mae_aux_disp

    tbar.update(1)

    line = [i, list_disp_gt[i], rmse_aux_disp, mae_aux_disp, rmse_aux_depth, mae_aux_depth]
    write_report(report_name, line)

tbar.close()

total_rmse_disp = global_rmse_disp / len(list_disp_est)
total_mae_disp = global_mae_disp / len(list_disp_est)

total_rmse_depth = global_rmse_depth / len(list_disp_est)
total_mae_depth = global_mae_depth / len(list_disp_est)

write_report(report_name, [data_path_est])
write_report(report_name, ["disp RMSE:", total_rmse_disp])
write_report(report_name, ["disp MAE: ", total_mae_disp])
write_report(report_name, ["depth RMSE: ", total_rmse_depth])
write_report(report_name, ["depth MAE: ", total_mae_depth])
Exemple #2
0
    disp_est[(mask * (-1) + 1).astype(np.bool)] = 0
    disp_gt[(mask * (-1) + 1).astype(np.bool)] = 0

    disp_error = np.abs(disp_gt - disp_est)

    disp_range_1 = disp_error >= 0.1
    disp_range_2 = disp_error >= 0.5
    disp_range_3 = disp_error >= 1

    global_disp_range_1 += np.count_nonzero(disp_range_1) / (disp_range_1.size)
    global_disp_range_2 += np.count_nonzero(disp_range_2) / (disp_range_2.size)
    global_disp_range_3 += np.count_nonzero(disp_range_3) / (disp_range_3.size)

    tbar.update(1)

    line = [i, list_disp_gt[i], np.count_nonzero(disp_range_1) / (disp_range_1.size),
            np.count_nonzero(disp_range_2) / (disp_range_2.size),
            np.count_nonzero(disp_range_3) / (disp_range_3.size)]
    write_report(report_name, line)

tbar.close()

total_dis_error_range_1 = global_disp_range_1 / len(list_disp_gt)
total_dis_error_range_2 = global_disp_range_2 / len(list_disp_gt)
total_dis_error_range_3 = global_disp_range_3 / len(list_disp_gt)

write_report(report_name, [report_name])
write_report(report_name, ["disp >0.1:", total_dis_error_range_1])
write_report(report_name, ["disp >0.5: ", total_dis_error_range_2])
write_report(report_name, ["disp >1: ", total_dis_error_range_3])
    disp_error = np.abs(disp_gt - disp_est)

    disp_range_1 = disp_error >= 0.1
    disp_range_2 = disp_error >= 0.5
    disp_range_3 = disp_error >= 1

    global_disp_range_1 += np.count_nonzero(disp_range_1) / (disp_range_1.size)
    global_disp_range_2 += np.count_nonzero(disp_range_2) / (disp_range_2.size)
    global_disp_range_3 += np.count_nonzero(disp_range_3) / (disp_range_3.size)

    tbar.update(1)

    line = [
        i, list_disp_gt[i],
        np.count_nonzero(disp_range_1) / (disp_range_1.size),
        np.count_nonzero(disp_range_2) / (disp_range_2.size),
        np.count_nonzero(disp_range_3) / (disp_range_3.size)
    ]
    write_report(report_name, line)

tbar.close()

total_dis_error_range_1 = global_disp_range_1 / len(list_disp_est)
total_dis_error_range_2 = global_disp_range_2 / len(list_disp_est)
total_dis_error_range_3 = global_disp_range_3 / len(list_disp_est)

write_report(report_name, [data_path_est])
write_report(report_name, ["disp >0.1:", total_dis_error_range_1])
write_report(report_name, ["disp >0.5: ", total_dis_error_range_2])
write_report(report_name, ["disp >1: ", total_dis_error_range_3])
Exemple #4
0
    depth_gt[(mask * (-1) + 1).astype(np.bool)] = 0
    depth_est[(mask * (-1) + 1).astype(np.bool)] = 0

    rmse_aux_depth = rmse(depth_gt, depth_est, mask)
    mae_aux_depth = mae(depth_gt, depth_est, mask)

    global_rmse_depth += rmse_aux_depth
    global_mae_depth += mae_aux_depth

    global_rmse_disp += rmse_aux_disp
    global_mae_disp += mae_aux_disp

    tbar.update(1)

    line = [i, list_disp_gt[i], rmse_aux_disp, mae_aux_disp, rmse_aux_depth, mae_aux_depth]
    write_report(report_name, line)

tbar.close()

total_rmse_disp = global_rmse_disp / len(list_disp_gt)
total_mae_disp = global_mae_disp / len(list_disp_gt)

total_rmse_depth = global_rmse_depth / len(list_disp_gt)
total_mae_depth = global_mae_depth / len(list_disp_gt)

write_report(report_name, [report_name])
write_report(report_name, ["disp RMSE:", total_rmse_disp])
write_report(report_name, ["disp MAE: ", total_mae_disp])
write_report(report_name, ["depth RMSE: ", total_rmse_depth])
write_report(report_name, ["depth MAE: ", total_mae_depth])