More reasonable speeds

Author: codeSamuraii

README.md

@@ -19,7 +19,15 @@ for frame in extractor:
     process(frame)
 ```
 
-## Development
+## Installation
+
+> **PyPI package coming very soon!**
+
+```bash
+git clone https://github.com/codeSamuraii/viteo
+cd viteo
+pip install -v -e .
+```
 
 Building/rebuilding the package in editable mode:
 ```bash

src/native/src/frame_extractor.mm

@@ -99,6 +99,11 @@ bool setupReader(int64_t startFrame) {
                 (id)kCVPixelBufferPixelFormatTypeKey: @(kCVPixelFormatType_32BGRA),
                 (id)kCVPixelBufferMetalCompatibilityKey: @YES,
                 (id)kCVPixelBufferIOSurfacePropertiesKey: @{},
+                // Add VideoToolbox hardware acceleration hints
+                AVVideoDecompressionPropertiesKey: @{
+                    (id)kVTDecompressionPropertyKey_UsingHardwareAcceleratedVideoDecoder: @YES,
+                    (id)kVTDecompressionPropertyKey_PropagatePerFrameHDRDisplayMetadata: @NO,
+                },
             };
 
             output = [[AVAssetReaderTrackOutput alloc]

tests/test_viteo.py

@@ -29,25 +29,25 @@ def video_files(test_data_dir):
             "path": test_data_dir / "video_4k.mp4",
             "width": 3840,
             "height": 2160,
-            "min_fps": 220.0  # Minimum expected extraction speed
+            "min_fps": 60.0  # Minimum expected extraction speed
         },
         "1080p": {
             "path": test_data_dir / "video_1080p.mp4",
             "width": 1920,
             "height": 1080,
-            "min_fps": 630.0
+            "min_fps": 120.0
         },
         "720p": {
             "path": test_data_dir / "video_720p.mp4",
             "width": 1280,
             "height": 720,
-            "min_fps": 1080.0
+            "min_fps": 240.0
         },
         "480p": {
             "path": test_data_dir / "video_480p.mp4",
             "width": 854,
             "height": 480,
-            "min_fps": 1700.0
+            "min_fps": 480.0
         }
     }