using namespace std;
+double eval_flow(const char *filename1, const char *filename2);
+
int main(int argc, char **argv)
{
- Flow flow = read_flow(argv[1]);
- Flow gt = read_flow(argv[2]);
+ double sum_epe = 0.0;
+ int num_flows = 0;
+ for (int i = 1; i < argc; i += 2) {
+ sum_epe += eval_flow(argv[i], argv[i + 1]);
+ ++num_flows;
+ }
+ printf("Average EPE: %.2f pixels\n", sum_epe / num_flows);
+}
+
+double eval_flow(const char *filename1, const char *filename2)
+{
+ Flow flow = read_flow(filename1);
+ Flow gt = read_flow(filename2);
double sum = 0.0;
for (unsigned y = 0; y < unsigned(flow.height); ++y) {
sum += hypot(du - gt_du, dv - gt_dv);
}
}
- fprintf(stderr, "Average EPE: %.2f pixels\n", sum / (flow.width * flow.height));
+ return sum / (flow.width * flow.height);
}