annotation bug fix

MacherLabs · Mar 25, 2020 · 887a1c3 · 887a1c3
1 parent a20b2ca
commit 887a1c3
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/common/videoRecorder.py b/common/videoRecorder.py
@@ -35,16 +35,19 @@ def __init__(self, filename, fourcc, fps, frameSize=None,limit_fps=False, base_d
 
         if timestamp_filename:
             self.filename_pattern = '%s-%s%s'%(self.filename, '{}', self.file_extension)
-            self.filename_pattern_annotations = '%s-%s%s'%(self.filename, '{}', self.annotations_extension)
+            if self.annotations:
+                self.filename_pattern_annotations = '%s-%s%s'%(self.filename, '{}', self.annotations_extension)
             self.cleanup_pattern = '.*%s-%s%s'%(self.filename, '.*', self.file_extension)
         else:
             self.filename_pattern = '%s%s'%(self.filename, self.file_extension)
-            self.filename_pattern_annotations = '%s-%s'%(self.filename, self.annotations_extension)
+            if self.annotations:
+                self.filename_pattern_annotations = '%s-%s'%(self.filename, self.annotations_extension)
             self.cleanup_pattern = '.*%s%s'%(self.filename, self.file_extension)
 
         if base_dir:
             self.filename_pattern = os.path.join(base_dir, self.filename_pattern)
-            self.filename_pattern_annotations = os.path.join(base_dir,self.filename_pattern_annotations)
+            if self.annotations:
+                self.filename_pattern_annotations = os.path.join(base_dir,self.filename_pattern_annotations)
             self.base_dir = base_dir
         else:
             self.base_dir = os.path.dirname(self.filename)