Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 2 additions & 4 deletions converter/gpu_error_correction.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,5 @@
"""
GPU-Accelerated Reed-Solomon Error Correction
High-performance CUDA implementation for NVIDIA GPUs
"""
"""GPU-accelerated Reed‑Solomon error correction using CUDA."""


import numpy as np
import cupy as cp
Expand Down
6 changes: 2 additions & 4 deletions converter/gpu_frame_generator.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,5 @@
"""
GPU-Accelerated Frame Generation Pipeline
Direct GPU memory operations for maximum throughput
"""
"""GPU-based frame generation pipeline optimized for high throughput."""


import numpy as np
import cupy as cp
Expand Down
26 changes: 26 additions & 0 deletions converter/video_raptor_encoder.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
"""Helpers for metadata and sync frame generation for Raptor videos."""

import numpy as np

class VideoRaptorEncoder:
"""Simple metadata and sync frame generator"""

def __init__(self, width: int, height: int, fps: int = 30):
self.width = width
self.height = height
self.fps = fps

def create_metadata_frame(self, file_info: dict) -> np.ndarray:
frame = np.zeros((self.height, self.width, 3), dtype=np.uint8)
return frame

def create_calibration_frame(self) -> np.ndarray:
frame = np.zeros((self.height, self.width, 3), dtype=np.uint8)
return frame

def _add_sync_pattern(self, frame: np.ndarray) -> None:
step_y = max(1, self.height // 20)
step_x = max(1, self.width // 20)
frame[::step_y, :] = 255
frame[:, ::step_x] = 255

5 changes: 5 additions & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -17,5 +17,10 @@ cupy-cuda12x>=12.0.0
# For CUDA 11.x (comment above and uncomment below):
# cupy-cuda11x>=11.0.0

cupy-cuda12x==13.2.0
numba==0.59.1
pyldpc==0.5.2
opencv-python>=4.9.0.0

# Error correction (removed reedsolo, using GPU Raptor codes instead)
# reedsolo>=1.5.4 # Replaced by GPU implementation
Loading