annotation bug fix cont..

MacherLabs · Mar 25, 2020 · 352f67e · 352f67e
1 parent 887a1c3
commit 352f67e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/common/videoRecorder.py b/common/videoRecorder.py
@@ -78,9 +78,10 @@ def write(self, imgcv, annotations = None):
             # Start new recording
             datestr = datetime.now().strftime("%d-%m-%y-%H%M%S")
             self.vname = self.filename_pattern.format(datestr)
-            self.aname=self.filename_pattern_annotations.format(datestr)
-            logger.info('Annotations file created - %s', self.aname)
-            self.writeAnnotationHeader()
+            if self.annotations:
+                self.aname=self.filename_pattern_annotations.format(datestr)
+                logger.info('Annotations file created - %s', self.aname)
+                self.writeAnnotationHeader()
             self.videow = cv2.VideoWriter(self.vname, self.fourcc, int(self.fps), self.frameSize)
             logger.info('Video Saved - %s', self.vname)
             # Frame Count and fps calculation variable init