Spaces:
Sleeping
Sleeping
File size: 2,250 Bytes
2978bba | 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 | #!/usr/bin/env python3
"""
Auto-test script for MorphGuard demo:
- Enumerate attached webcams (up to 4)
- Capture a frame from each camera
- Send each to the /api/detect endpoint
- Send the first frame to /api/demorph
"""
try:
import cv2
except ImportError:
print("Error: OpenCV (cv2) not installed. Run 'pip install opencv-python' and retry.")
exit(1)
import requests
import time
BASE_URL = 'http://localhost:5000'
def list_and_capture_cams(max_cams=4):
caps = []
for i in range(max_cams):
cap = cv2.VideoCapture(i)
if cap.isOpened():
caps.append((i, cap))
else:
cap.release()
frames = []
for idx, cap in caps:
ret, frame = cap.read()
cap.release()
if ret:
# Encode to JPEG bytes
_, buf = cv2.imencode('.jpg', frame)
frames.append((idx, buf.tobytes()))
return frames
def detect_images(frames):
print(f"Sending {len(frames)} captured camera frames to /api/detect...")
results = []
for idx, img_bytes in frames:
resp = requests.post(f"{BASE_URL}/api/detect", files={'image': (f'cam{idx}.jpg', img_bytes)})
try:
data = resp.json()
except ValueError:
data = {'error': 'invalid json', 'status_code': resp.status_code}
print(f"Camera {idx} detect response:", data)
results.append(data)
return results
def demorph_image(frame):
idx, img_bytes = frame
print(f"Sending camera {idx} frame to /api/demorph...")
resp = requests.post(f"{BASE_URL}/api/demorph", files={'image': (f'cam{idx}.jpg', img_bytes)})
try:
data = resp.json()
except ValueError:
data = {'error': 'invalid json', 'status_code': resp.status_code}
print(f"Demorph response:", data)
return data
def main():
print("Starting auto-test of MorphGuard demo...")
# Give server a moment
time.sleep(2)
frames = list_and_capture_cams()
if not frames:
print("No webcams detected. Exiting.")
return
detect_results = detect_images(frames)
# Demorph first detected frame
demorph_data = demorph_image(frames[0])
print("Auto-test complete.")
if __name__ == '__main__':
main() |