commit 3b815f1a5c55109bfd917ed4d038bfb13bd4cc17
author Austin Schuh <austin.linux@gmail.com> Fri Mar 08 20:10:37 2019 -0800
committer Austin Schuh <austin.linux@gmail.com> Fri Mar 08 21:18:00 2019 -0800
tree 1b2687c121ea83518f5c2bd45f277953e90a8938
parent 0fd1cef7c6df7c6058007a7421e53ea5ca2eb88b

Make the calibration image path a flag

Change-Id: I8f634aba86a5578bceab530b9f86b630927548a7

y2019/vision/global_calibration.cc

diff --git a/y2019/vision/global_calibration.cc b/y2019/vision/global_calibration.cc
index d269db5..88d657b 100644
--- a/y2019/vision/global_calibration.cc
+++ b/y2019/vision/global_calibration.cc
@@ -18,6 +18,7 @@
 #include "ceres/ceres.h"
 
 DEFINE_int32(camera_id, -1, "The camera ID to calibrate");
+DEFINE_string(prefix, "", "The image filename prefix");
 
 using ::aos::events::DataSocket;
 using ::aos::events::RXUdpSocket;
@@ -71,16 +72,14 @@
 
 void main(int argc, char **argv) {
   using namespace y2019::vision;
-  gflags::ParseCommandLineFlags(&argc, &argv, false);
-
-  const char *base_directory = "/home/parker/data/frc/2019_calibration/";
+  ::gflags::ParseCommandLineFlags(&argc, &argv, false);
 
   DatasetInfo info = {
       FLAGS_camera_id,
       {{12.5 * kInchesToMeters, 0.5 * kInchesToMeters}},
       {{kInchesToMeters, 0.0}},
       26,
-      "cam5_0/debug_viewer_jpeg_",
+      FLAGS_prefix.c_str(),
       59,
   };
 
@@ -114,9 +113,8 @@
   ::std::cout << "error = " << summary.final_cost << ";\n";
 
   for (int i = 0; i < info.num_images; ++i) {
-    auto frame = aos::vision::LoadFile(std::string(base_directory) +
-                                       info.filename_prefix +
-                                       std::to_string(i) + ".yuyv");
+    ::aos::vision::DatasetFrame frame =
+        aos::vision::LoadFile(FLAGS_prefix + std::to_string(i) + ".yuyv");
 
     const ::aos::vision::ImageFormat fmt{640, 480};
     ::aos::vision::BlobList imgs =