bug fix for audios

midas-research · May 30, 2024 · e2b4c07 · e2b4c07
1 parent 4cc294c
commit e2b4c07
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/cvat/apps/engine/media_extractors.py b/cvat/apps/engine/media_extractors.py
@@ -546,7 +546,10 @@ def _get_av_container(self):
             self._source_path[0].seek(0) # required for re-reading
 
         encoding = self.get_file_encoding(self._source_path[0])
-        return av.open(self._source_path[0], metadata_encoding = encoding)
+        if encoding:
+            return av.open(self._source_path[0], metadata_encoding = encoding)
+        else:
+            return av.open(self._source_path[0])
 
     def _get_duration(self):
         with self._get_av_container() as container:

diff --git a/cvat/apps/engine/task.py b/cvat/apps/engine/task.py
@@ -970,7 +970,10 @@ def get_audio_duration(file_path):
         slogger.glob.debug("ENCODING")
         slogger.glob.debug(encoding)
         # Open the audio file
-        container = av.open(file_path, metadata_encoding=encoding)
+        if encoding:
+            container = av.open(file_path, metadata_encoding=encoding)
+        else:
+            container = av.open(file_path)
 
         # Get the first audio stream
         audio_stream = next((stream for stream in container.streams if stream.codec.type == 'audio'), None)