Spaces:
Runtime error
Runtime error
File size: 4,461 Bytes
0469d65 |
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 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 |
#!/usr/bin/env python3
"""
High FPS test for SafetyMaster Pro
Tests the optimized detection pipeline for maximum performance
"""
import cv2
import time
from safety_detector import SafetyDetector
from camera_manager import CameraManager
def test_high_fps():
"""Test the system at high FPS."""
print("🚀 SafetyMaster Pro - High FPS Performance Test")
print("=" * 50)
# Initialize components
detector = SafetyDetector()
camera_manager = CameraManager(source=0)
if not camera_manager.start_capture():
print("❌ Failed to start camera")
return
print(f"📹 Camera started: {camera_manager.get_properties()}")
print("🎯 Testing high FPS performance...")
print(" Press 'q' to quit, 's' to save frame")
frame_count = 0
detection_count = 0
start_time = time.time()
last_detection_results = None
try:
while True:
frame_data = camera_manager.get_latest_frame()
if frame_data is not None:
frame, timestamp = frame_data
frame_count += 1
# Run detection every 3rd frame for optimal performance
if frame_count % 3 == 0 or last_detection_results is None:
detection_start = time.time()
results = detector.detect_safety_violations(frame)
detection_time = time.time() - detection_start
last_detection_results = results
detection_count += 1
else:
# Use cached results for intermediate frames
results = last_detection_results
detection_time = 0
# Draw detections
annotated_frame = detector.draw_detections(frame, results)
# Calculate and display FPS
elapsed_time = time.time() - start_time
if elapsed_time > 0:
video_fps = frame_count / elapsed_time
detection_fps = detection_count / elapsed_time
# Add FPS info to frame
cv2.putText(annotated_frame, f"Video FPS: {video_fps:.1f}",
(10, 30), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 255, 0), 2)
cv2.putText(annotated_frame, f"AI FPS: {detection_fps:.1f}",
(10, 60), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 255, 255), 2)
cv2.putText(annotated_frame, f"Detection Time: {detection_time*1000:.1f}ms",
(10, 90), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (255, 255, 0), 2)
# Display frame
cv2.imshow('SafetyMaster Pro - High FPS Test', annotated_frame)
# Print stats every 60 frames
if frame_count % 60 == 0:
print(f"📊 Frame {frame_count}: Video FPS: {video_fps:.1f}, AI FPS: {detection_fps:.1f}")
if results['violations']:
print(f" ⚠️ {len(results['violations'])} violations detected")
key = cv2.waitKey(1) & 0xFF
if key == ord('q'):
break
elif key == ord('s'):
# Save frame
filename = f"high_fps_test_{int(time.time())}.jpg"
cv2.imwrite(filename, annotated_frame)
print(f"💾 Saved {filename}")
else:
time.sleep(0.001) # Minimal delay when no frame available
except KeyboardInterrupt:
print("\n🛑 Test interrupted by user")
finally:
camera_manager.stop_capture()
cv2.destroyAllWindows()
# Final statistics
total_time = time.time() - start_time
print(f"\n📈 Final Performance Statistics:")
print(f" Total frames: {frame_count}")
print(f" Total detections: {detection_count}")
print(f" Test duration: {total_time:.1f}s")
print(f" Average video FPS: {frame_count / total_time:.1f}")
print(f" Average AI FPS: {detection_count / total_time:.1f}")
print(f" Frame skip ratio: {(frame_count - detection_count) / frame_count * 100:.1f}%")
if __name__ == "__main__":
test_high_fps() |