removed alpha channel from video frames

vitrivr · Sep 11, 2023 · d6d4c3f · d6d4c3f
1 parent 8987321
commit d6d4c3f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cineast-core/src/main/java/org/vitrivr/cineast/core/features/FESVectorFeature.java b/cineast-core/src/main/java/org/vitrivr/cineast/core/features/FESVectorFeature.java
@@ -414,6 +414,7 @@ private float[] embedModel(IModel model) throws IOException, InterruptedExceptio
 
   private float[] embedVideo(List<MultiImage> frames) throws IOException, InterruptedException {
     var images = frames.stream().map(image -> image.getBufferedImage()).collect(Collectors.toList());
+    images = images.stream().map(img -> new BufferedImage(img.getWidth(), img.getHeight(), BufferedImage.TYPE_INT_RGB)).collect(Collectors.toList());
     var encodings = embedMultipleImages(images);
 
     // Sum