nedo-vision-worker-core 0.4.0__tar.gz → 0.4.1__tar.gz

{nedo_vision_worker_core-0.4.0 → nedo_vision_worker_core-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker-core
-Version: 0.4.0
+Version: 0.4.1
 Summary: Nedo Vision Worker Core Library for AI Vision Processing
 Author-email: Willy Achmat Fauzi <willy.achmat@gmail.com>
 Maintainer-email: Willy Achmat Fauzi <willy.achmat@gmail.com>

{nedo_vision_worker_core-0.4.0 → nedo_vision_worker_core-0.4.1}/nedo_vision_worker_core/__init__.py

@@ -7,10 +7,10 @@ A library for running AI vision processing and detection in the Nedo Vision plat
 from .core_service import CoreService
 from .callbacks import DetectionType, CallbackTrigger, DetectionData, IntervalMetadata
 
-__version__ = "0.4.0"
+__version__ = "0.4.1"
 __all__ = [
     "CoreService", 
-    "DetectionType", 
+    "DetectionType",
     "CallbackTrigger", 
     "DetectionData",
     "IntervalMetadata",

{nedo_vision_worker_core-0.4.0 → nedo_vision_worker_core-0.4.1}/nedo_vision_worker_core/pipeline/PipelineProcessor.py

@@ -259,9 +259,10 @@ class PipelineProcessor:
                     finally:
                         self.rtmp_streamer = None
 
-                # Simplified queue feeding - avoid expensive try-except
-                if not self.frame_queue.full():
-                    self.frame_queue.put_nowait(frame)
+                # Only feed frames to detection queue if detection processor is active
+                if self.detection_processor is not None:
+                    if not self.frame_queue.full():
+                        self.frame_queue.put_nowait(frame)
                 
                 loop_elapsed = time.time() - loop_start
                 sleep_time = max(0.001, target_frame_time - loop_elapsed)
@@ -307,7 +308,13 @@ class PipelineProcessor:
 
         while self.running:
             try:
-                frame = self.frame_queue.get(block=True, timeout=1)
+                # Calculate how long to wait for next detection
+                current_time = time.time()
+                time_since_last_detection = current_time - last_detection_time
+                time_until_next_detection = max(0.1, self.detection_interval - time_since_last_detection)
+                
+                # Wait for frame with timeout aligned to detection interval
+                frame = self.frame_queue.get(block=True, timeout=time_until_next_detection)
                 
                 # Check for poison pill (None = stop signal)
                 if frame is None:
@@ -332,9 +339,14 @@ class PipelineProcessor:
                 except queue.Empty:
                     pass
 
-                # Respect detection interval
+                # Respect detection interval - skip if too soon
                 if (current_time - last_detection_time) < self.detection_interval:
+                    # Sleep the remaining time instead of busy-waiting
+                    remaining_time = self.detection_interval - (current_time - last_detection_time)
+                    if remaining_time > 0.01:
+                        time.sleep(remaining_time)
                     continue
+                
                 last_detection_time = current_time
 
                 if self.detection_processor is None or frame is None or frame.size == 0:
@@ -363,6 +375,7 @@ class PipelineProcessor:
                     )
 
             except queue.Empty:
+                # Timeout occurred - this is normal, just continue to next iteration
                 pass
             except Exception as e:
                 logging.error(f"❌ Error in detection thread for pipeline {pipeline_id}: {e}", exc_info=True)

{nedo_vision_worker_core-0.4.0 → nedo_vision_worker_core-0.4.1}/nedo_vision_worker_core/streams/RTMPStreamer.py

@@ -307,6 +307,7 @@ class RTMPStreamer:
                 if not self.is_active():
                     raise BrokenPipeError("FFmpeg process is not active")
                 self._ffmpeg_process.stdin.write(frame.tobytes())
+                self._ffmpeg_process.stdin.flush()  # Ensure data is sent immediately
             except (BrokenPipeError, OSError) as e:
                 
                 # Check if this failure was from a HW encoder
@@ -356,23 +357,20 @@ class RTMPStreamer:
             '-framerate', str(self.fps), '-i', '-',
         ]
         
+        # Add encoder and encoder-specific parameters
         cmd.extend(encoder_args)
         
+        # Common video parameters
         cmd.extend([
-            '-profile:v', 'main', '-pix_fmt', 'yuv420p',
+            '-pix_fmt', 'yuv420p',
             '-b:v', f"{self.bitrate}k", '-maxrate', f"{self.bitrate}k", '-bufsize', f"{self.bitrate*2}k",
             '-g', str(self.fps * 2), '-keyint_min', str(self.fps),
-            '-force_key_frames', 'expr:gte(t,n_forced*1)', '-an',
-            '-flvflags', 'no_duration_filesize', '-f', 'flv', self.rtmp_url,
+            '-force_key_frames', 'expr:gte(t,n_forced*1)', 
+            '-an',  # No audio
+            '-flvflags', 'no_duration_filesize', 
+            '-f', 'flv', self.rtmp_url,
         ])
         
-        if encoder_name == "libx264":
-            cmd.extend([
-                "-preset", "ultrafast",
-                "-tune", "zerolatency",
-                "-x264-params", "open_gop=0:aud=1:repeat-headers=1:nal-hrd=cbr",
-            ])
-        
         return cmd, encoder_name
 
     def _select_ffmpeg_encoder(self, force_cpu: bool = False) -> Tuple[List[str], str]:
@@ -381,26 +379,74 @@ class RTMPStreamer:
         Will force CPU if force_cpu is True.
         """
         if force_cpu:
-            return ["-c:v", "libx264"], "libx264"
+            return [
+                "-c:v", "libx264",
+                "-preset", "ultrafast",
+                "-tune", "zerolatency",
+                "-profile:v", "main",
+            ], "libx264"
         
         force_encoder = os.environ.get("RTMP_ENCODER", "").lower()
         
         if force_encoder == "cpu" or force_encoder == "libx264":
-            return ["-c:v", "libx264"], "libx264"
+            return [
+                "-c:v", "libx264",
+                "-preset", "ultrafast",
+                "-tune", "zerolatency",
+                "-profile:v", "main",
+            ], "libx264"
         elif force_encoder == "nvenc":
-            return ["-c:v", "h264_nvenc", "-preset", "llhp"], "h264_nvenc"
+            return [
+                "-c:v", "h264_nvenc",
+                "-preset", "p1",  # p1 = fastest, p7 = slowest
+                "-tune", "ull",  # ultra-low latency
+                "-rc:v", "cbr",  # constant bitrate for streaming
+                "-rc-lookahead", "0",  # disable lookahead for lower latency
+                "-delay", "0",  # zero delay
+                "-zerolatency", "1",  # enable zero latency mode
+                "-profile:v", "main",
+                "-gpu", "0",  # Use first GPU
+            ], "h264_nvenc"
         
         if self._platform.is_jetson():
-            # Jetson-specific encoder
-            return ["-c:v", "h264_omx"], "h264_omx"
+            # Jetson-specific encoder with optimizations
+            return [
+                "-c:v", "h264_nvenc",
+                "-preset", "p1",
+                "-tune", "ull",
+                "-rc:v", "cbr",
+                "-rc-lookahead", "0",
+                "-delay", "0",
+                "-zerolatency", "1",
+                "-profile:v", "main",
+            ], "h264_nvenc"
 
         if sys.platform == "darwin": 
-            return ["-c:v", "h264_videotoolbox"], "h264_videotoolbox"
+            return [
+                "-c:v", "h264_videotoolbox",
+                "-profile:v", "main",
+                "-realtime", "1",
+            ], "h264_videotoolbox"
         
         has_nvidia = (os.environ.get("NVIDIA_VISIBLE_DEVICES") is not None or 
                       os.path.exists("/proc/driver/nvidia/version"))
         
         if has_nvidia:
-            return ["-c:v", "h264_nvenc", "-preset", "llhp"], "h264_nvenc"
+            return [
+                "-c:v", "h264_nvenc",
+                "-preset", "p1",  # p1 = fastest preset
+                "-tune", "ull",  # ultra-low latency
+                "-rc:v", "cbr",  # constant bitrate
+                "-rc-lookahead", "0",  # disable lookahead
+                "-delay", "0",  # zero delay
+                "-zerolatency", "1",  # zero latency mode
+                "-profile:v", "main",
+                "-gpu", "0",  # Use first GPU
+            ], "h264_nvenc"
         
-        return ["-c:v", "libx264"], "libx264"
+        return [
+            "-c:v", "libx264",
+            "-preset", "ultrafast",
+            "-tune", "zerolatency",
+            "-profile:v", "main",
+        ], "libx264"

{nedo_vision_worker_core-0.4.0 → nedo_vision_worker_core-0.4.1}/nedo_vision_worker_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker-core
-Version: 0.4.0
+Version: 0.4.1
 Summary: Nedo Vision Worker Core Library for AI Vision Processing
 Author-email: Willy Achmat Fauzi <willy.achmat@gmail.com>
 Maintainer-email: Willy Achmat Fauzi <willy.achmat@gmail.com>