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])
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])
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])