Merge pull request #334 from OpenTrafficCam/feature/2434-show-backgro…

…und-image-in-grayscale feature/2434-show-background-image-in-grayscale
OpenTrafficCam · Sep 21, 2023 · c791c9c · c791c9c
2 parents 0cd9b3f + b54292d
commit c791c9c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/OTAnalytics/plugin_video_processing/video_reader.py b/OTAnalytics/plugin_video_processing/video_reader.py
@@ -6,6 +6,8 @@
 from OTAnalytics.domain.track import PilImage, TrackImage
 from OTAnalytics.domain.video import VideoReader
 
+GRAYSCALE = "L"
+
 
 class InvalidVideoError(Exception):
     pass
@@ -41,4 +43,4 @@ def get_frame(self, video_path: Path, index: int) -> TrackImage:
         clip.close()
         if found is None:
             raise FrameDoesNotExistError(f"frame number '{index}' does not exist")
-        return PilImage(Image.fromarray(found))
+        return PilImage(Image.fromarray(found).convert(GRAYSCALE))