fixing #219

pupil-labs · Oct 30, 2015 · 70ae734 · 70ae734
1 parent fcb0f5b
commit 70ae734
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pupil_src/shared_modules/calibration_routines/camera_intrinsics_estimation.py b/pupil_src/shared_modules/calibration_routines/camera_intrinsics_estimation.py
@@ -108,7 +108,7 @@ def get_count(self):
         return self.count
 
     def advance(self,_):
-        if self.count ==10:
+        if self.count == 10:
             logger.info("Capture 10 calibration patterns.")
             self.button.status_text = "%i to go" %(self.count)
 
@@ -178,6 +178,8 @@ def calculate(self):
         camera_calibration = {'camera_matrix':camera_matrix,'dist_coefs':dist_coefs,'camera_name':self.g_pool.capture.name,'resolution':self.g_pool.capture.frame_size}
         save_object(camera_calibration,os.path.join(self.g_pool.user_dir,"camera_calibration"))
         logger.info("Calibration saved to user folder")
+        self.count = 10
+
 
     def update(self,frame,events):
         if self.collect_new: