streaming part1

test_streaming.py

@@ -0,0 +1,142 @@
+#!/usr/bin/env python3
+"""
+Test script to verify streaming implementation components
+"""
+
+import sys
+from pathlib import Path
+
+def test_imports():
+    """Test that all modules can be imported"""
+    print("Testing imports...")
+    
+    try:
+        from vr180_streaming import VR180StreamingProcessor, StreamingConfig
+        print("✅ Main imports successful")
+    except ImportError as e:
+        print(f"❌ Failed to import main modules: {e}")
+        return False
+        
+    try:
+        from vr180_streaming.frame_reader import StreamingFrameReader
+        from vr180_streaming.frame_writer import StreamingFrameWriter
+        from vr180_streaming.stereo_manager import StereoConsistencyManager
+        from vr180_streaming.sam2_streaming import SAM2StreamingProcessor
+        from vr180_streaming.detector import PersonDetector
+        print("✅ Component imports successful")
+    except ImportError as e:
+        print(f"❌ Failed to import components: {e}")
+        return False
+        
+    return True
+
+def test_config():
+    """Test configuration loading"""
+    print("\nTesting configuration...")
+    
+    try:
+        from vr180_streaming.config import StreamingConfig
+        
+        # Test creating config
+        config = StreamingConfig()
+        print("✅ Config creation successful")
+        
+        # Test config validation
+        errors = config.validate()
+        print(f"   Config errors: {len(errors)} (expected, no paths set)")
+        
+        return True
+    except Exception as e:
+        print(f"❌ Config test failed: {e}")
+        return False
+
+def test_dependencies():
+    """Test required dependencies"""
+    print("\nTesting dependencies...")
+    
+    deps_ok = True
+    
+    # Test PyTorch
+    try:
+        import torch
+        print(f"✅ PyTorch {torch.__version__}")
+        if torch.cuda.is_available():
+            print(f"   CUDA available: {torch.cuda.get_device_name(0)}")
+        else:
+            print("   ⚠️  CUDA not available")
+    except ImportError:
+        print("❌ PyTorch not installed")
+        deps_ok = False
+        
+    # Test OpenCV
+    try:
+        import cv2
+        print(f"✅ OpenCV {cv2.__version__}")
+    except ImportError:
+        print("❌ OpenCV not installed")
+        deps_ok = False
+        
+    # Test Ultralytics
+    try:
+        from ultralytics import YOLO
+        print("✅ Ultralytics YOLO available")
+    except ImportError:
+        print("❌ Ultralytics not installed")
+        deps_ok = False
+        
+    # Test other deps
+    try:
+        import yaml
+        import numpy as np
+        import psutil
+        print("✅ Other dependencies available")
+    except ImportError as e:
+        print(f"❌ Missing dependency: {e}")
+        deps_ok = False
+        
+    return deps_ok
+
+def test_frame_reader():
+    """Test frame reader with a dummy video"""
+    print("\nTesting StreamingFrameReader...")
+    
+    try:
+        from vr180_streaming.frame_reader import StreamingFrameReader
+        
+        # Would need an actual video file to test
+        print("⚠️  Skipping reader test (no test video)")
+        return True
+        
+    except Exception as e:
+        print(f"❌ Frame reader test failed: {e}")
+        return False
+
+def main():
+    """Run all tests"""
+    print("🧪 VR180 Streaming Implementation Test")
+    print("=" * 40)
+    
+    all_ok = True
+    
+    # Run tests
+    all_ok &= test_imports()
+    all_ok &= test_config()
+    all_ok &= test_dependencies()
+    all_ok &= test_frame_reader()
+    
+    print("\n" + "=" * 40)
+    if all_ok:
+        print("✅ All tests passed!")
+        print("\nNext steps:")
+        print("1. Install SAM2: cd segment-anything-2 && pip install -e .")
+        print("2. Download checkpoints: cd checkpoints && ./download_ckpts.sh")
+        print("3. Create config: python -m vr180_streaming --generate-config my_config.yaml")
+        print("4. Run processing: python -m vr180_streaming my_config.yaml")
+    else:
+        print("❌ Some tests failed")
+        print("\nPlease run: pip install -r requirements.txt")
+        
+    return 0 if all_ok else 1
+
+if __name__ == "__main__":
+    sys.exit(main())