Spaces:
Sleeping
Sleeping
File size: 2,359 Bytes
e40d5b1 |
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 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
"""
Create a simple test video with lane-like features for testing
"""
import cv2
import numpy as np
import os
def create_test_video(output_path, duration_sec=5, fps=30):
"""
Create a test video with simulated road and lanes
"""
width, height = 640, 480
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
total_frames = duration_sec * fps
for frame_idx in range(total_frames):
# Create dark gray background (asphalt)
frame = np.ones((height, width, 3), dtype=np.uint8) * 50
# Add some texture/noise for realism
noise = np.random.randint(0, 20, (height, width, 3), dtype=np.uint8)
frame = cv2.add(frame, noise)
# Draw road perspective trapezoid
road_points = np.array([
[int(width * 0.1), height],
[int(width * 0.4), int(height * 0.5)],
[int(width * 0.6), int(height * 0.5)],
[int(width * 0.9), height]
], dtype=np.int32)
cv2.fillPoly(frame, [road_points], (60, 60, 60))
# Calculate lane positions with slight animation
offset = int(10 * np.sin(frame_idx / 10))
# Left lane
left_bottom = (int(width * 0.3) + offset, height)
left_top = (int(width * 0.45) + offset, int(height * 0.6))
cv2.line(frame, left_bottom, left_top, (255, 255, 255), 3)
# Right lane
right_bottom = (int(width * 0.7) + offset, height)
right_top = (int(width * 0.55) + offset, int(height * 0.6))
cv2.line(frame, right_bottom, right_top, (255, 255, 255), 3)
# Add dashed center line
for y in range(height, int(height * 0.6), -30):
if (y // 30) % 2 == 0:
x_start = int(width * 0.5) + offset
x_end = int(width * 0.5) + offset
cv2.line(frame, (x_start, y), (x_end, y - 20), (255, 255, 0), 2)
# Write frame
out.write(frame)
out.release()
print(f"✓ Test video created: {output_path}")
print(f" Duration: {duration_sec}s, FPS: {fps}, Frames: {total_frames}")
if __name__ == "__main__":
# Create test video in /tmp
output_path = "/tmp/test_road_video.mp4"
create_test_video(output_path, duration_sec=3, fps=30)
|