sciveo 0.1.35__tar.gz → 0.1.36__tar.gz

{sciveo-0.1.35 → sciveo-0.1.36}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sciveo
-Version: 0.1.35
+Version: 0.1.36
 Description-Content-Type: text/markdown
 Provides-Extra: mon
 Provides-Extra: net

{sciveo-0.1.35 → sciveo-0.1.36}/sciveo/media/pipelines/processors/image/embeddings.py

@@ -21,7 +21,7 @@ from sciveo.tools.logger import *
 from sciveo.tools.common import *
 from sciveo.media.pipelines.processors.tpu_base import *
 from sciveo.media.pipelines.base import ApiContent
-from sciveo.nlp.images.embeddings import ImageEmbedding
+from sciveo.ml.images.embeddings import ImageEmbedding
 
 
 class ImageEmbeddingProcessor(TPUBaseProcessor):

sciveo-0.1.36/sciveo/ml/video/description.py

@@ -0,0 +1,41 @@
+#
+# Pavlin Georgiev, Softel Labs
+#
+# This is a proprietary file and may not be copied,
+# distributed, or modified without express permission
+# from the owner. For licensing inquiries, please
+# contact pavlin@softel.bg.
+#
+# 2024
+#
+
+import os
+import cv2
+import gc
+from PIL import Image
+
+import torch
+from transformers import AutoProcessor, AutoModelForCausalLM
+
+from sciveo.tools.common import *
+from sciveo.ml.images.description import ImageToText
+
+
+class VideoToText(ImageToText):
+  def __init__(self, model_id, max_length=64, cache_dir=None, device=None) -> None:
+    super().__init__(model_id, max_length, cache_dir, device)
+
+  def read_video_frame(self, video_path):
+    cap = cv2.VideoCapture(video_path)
+    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    cap.set(cv2.CAP_PROP_POS_FRAMES, total_frames // 2)
+    ret, frame = cap.read()
+    cap.release()
+    return frame
+
+  def predict(self, X):
+    frames = []
+    for x in X:
+      frames.append(self.read_video_frame(x))
+    predictions = super().predict(frames)
+    return predictions

sciveo-0.1.36/sciveo/version.py

@@ -0,0 +1,2 @@
+
+__version__ = '0.1.36'

{sciveo-0.1.35 → sciveo-0.1.36}/sciveo.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sciveo
-Version: 0.1.35
+Version: 0.1.36
 Description-Content-Type: text/markdown
 Provides-Extra: mon
 Provides-Extra: net

{sciveo-0.1.35 → sciveo-0.1.36}/sciveo.egg-info/SOURCES.txt

@@ -100,6 +100,8 @@ sciveo/ml/images/tools.py
 sciveo/ml/images/transforms.py
 sciveo/ml/nlp/__init__.py
 sciveo/ml/nlp/embeddings.py
+sciveo/ml/video/__init__.py
+sciveo/ml/video/description.py
 sciveo/monitoring/__init__.py
 sciveo/monitoring/monitor.py
 sciveo/monitoring/start.py

sciveo-0.1.35/sciveo/version.py

@@ -1,2 +0,0 @@
-
-__version__ = '0.1.35'