aegis-net/analyze_video.py at main · TherealArithmeticProgression/aegis-net · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
"""
Video Analysis Example - Process drone video with landing spot detection
"""
from agent import LandingSpotDetector

def analyze_video_with_best_settings():
    """
    Demonstrates video analysis with optimized settings
    """
    print("=" * 70)
    print("VIDEO LANDING SPOT ANALYSIS")
    print("=" * 70)

    # Create detector with optimal grid size
    detector = LandingSpotDetector(grid_size=15)

    # List available videos
    videos = [
        "videos/Bluemlisalphutte Flyover.mp4",
        "videos/Creux du Van Flight.mp4",
        "videos/Stockflue Flyaround.mp4",
    ]

    print("\nAvailable videos:")
    for i, video in enumerate(videos, 1):
        print(f"  {i}. {video.split('/')[-1]}")

    print("\nAnalyzing video with landing spot detection...")
    print("Two windows will appear:")
    print("  1. Original Drone Footage - Raw video feed")
    print("  2. Landing Spot Heatmap - Safety analysis overlay")
    print("\nControls:")
    print("  - Press 'q' to quit")
    print("  - Press 's' to save current frame analysis")
    print("-" * 70)

    # Process video (every 20th frame for smooth visualization)
    # Green areas = safe landing spots
    # Red areas = unsafe landing spots
    detector.process_video(videos[0], sample_rate=20)

    print("\nVideo analysis complete!")

if __name__ == "__main__":
    analyze_video_with_best_settings()