add main

README.md

@@ -93,16 +93,15 @@ git clone <repository-url>
 cd sam2e
 ./runpod_setup.sh
 
-# Then use the convenience scripts:
+# Then run with Python directly:
-./run_streaming.sh   # For streaming approach (recommended)
+python -m vr180_streaming config-streaming-runpod.yaml   # Streaming (recommended)
-./run_chunked.sh     # For chunked approach
+python -m vr180_matting config-chunked-runpod.yaml       # Chunked (original)
 ```
 
 The setup script will:
 - Install all dependencies
 - Download SAM2 models
-- Create example configs
+- Create example configs for both approaches
-- Set up convenience scripts
 
 ## Configuration
 

runpod_setup.sh

@@ -159,28 +159,7 @@ if [ ! -f "config-streaming-runpod.yaml" ]; then
     print_error "config-streaming-runpod.yaml not found - please check the repository"
 fi
 
-# Create convenience run scripts
+# Skip creating convenience scripts - use Python directly
-print_status "Creating run scripts..."
-
-# Chunked approach
-cat > run_chunked.sh << 'EOF'
-#!/bin/bash
-# Run VR180 matting with chunked approach (original)
-echo "🎬 Running VR180 matting - Chunked Approach"
-echo "==========================================="
-python -m vr180_matting.main config-chunked-runpod.yaml "$@"
-EOF
-chmod +x run_chunked.sh
-
-# Streaming approach
-cat > run_streaming.sh << 'EOF'
-#!/bin/bash
-# Run VR180 matting with streaming approach (optimized)
-echo "🎬 Running VR180 matting - Streaming Approach"
-echo "============================================="
-python -m vr180_streaming.main config-streaming-runpod.yaml "$@"
-EOF
-chmod +x run_streaming.sh
 
 # Test installation
 print_status "Testing installation..."
@@ -246,12 +225,10 @@ echo
 echo "2. Choose your processing approach:"
 echo
 echo "   a) STREAMING (Recommended - 2-3x faster, constant memory):"
-echo "      ./run_streaming.sh"
+echo "      python -m vr180_streaming config-streaming-runpod.yaml"
-echo "      # Or: python -m vr180_streaming config-streaming-runpod.yaml"
 echo
 echo "   b) CHUNKED (Original - more stable, higher memory):"
-echo "      ./run_chunked.sh"
+echo "      python -m vr180_matting config-chunked-runpod.yaml"
-echo "      # Or: python -m vr180_matting config-chunked-runpod.yaml"
 echo
 echo "3. Optional: Edit configs first:"
 echo "   nano config-streaming-runpod.yaml  # For streaming"
@@ -273,12 +250,12 @@ echo
 echo "🎯 Example Commands:"
 echo "==================="
 echo "# Process with custom output path:"
-echo "./run_streaming.sh --output /workspace/output/my_video.mp4"
+echo "python -m vr180_streaming config-streaming-runpod.yaml --output /workspace/output/my_video.mp4"
 echo
 echo "# Process specific frame range:"
-echo "./run_streaming.sh --start-frame 1000 --max-frames 5000"
+echo "python -m vr180_streaming config-streaming-runpod.yaml --start-frame 1000 --max-frames 5000"
 echo
 echo "# Override scale for quality:"
-echo "./run_streaming.sh --scale 0.75"
+echo "python -m vr180_streaming config-streaming-runpod.yaml --scale 0.75"
 echo
 echo "Happy matting! 🎬"

vr180_streaming/__main__.py

@@ -0,0 +1,9 @@
+#!/usr/bin/env python3
+"""
+VR180 Streaming entry point for python -m vr180_streaming
+"""
+
+from .main import main
+
+if __name__ == "__main__":
+    main()