219 lines
7.8 KiB
Python
219 lines
7.8 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
初始化示例数据脚本
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
from core.database import SessionLocal
|
||
from models.algorithm import Algorithm
|
||
from models.device import Device
|
||
from models.event import Event
|
||
from datetime import datetime, timedelta
|
||
import json
|
||
|
||
def init_sample_data():
|
||
"""初始化示例数据"""
|
||
db = SessionLocal()
|
||
|
||
try:
|
||
# 创建示例算法
|
||
algorithms = [
|
||
{
|
||
"name": "YOLOv11n人员检测",
|
||
"description": "基于YOLOv11n的人员检测算法，适用于边检场景",
|
||
"version": "1.0.0",
|
||
"model_path": "/models/yolo11n.pt",
|
||
"config_path": "/configs/yolo11n.yaml",
|
||
"status": "active",
|
||
"accuracy": 0.95,
|
||
"detection_classes": json.dumps(["person"]),
|
||
"input_size": "640x640",
|
||
"inference_time": 15.5,
|
||
"is_enabled": True,
|
||
"creator": "admin",
|
||
"tags": json.dumps(["person", "detection", "yolo"])
|
||
},
|
||
{
|
||
"name": "车辆检测算法",
|
||
"description": "专门用于车辆检测的深度学习算法",
|
||
"version": "2.1.0",
|
||
"model_path": "/models/vehicle_detection.pt",
|
||
"config_path": "/configs/vehicle.yaml",
|
||
"status": "active",
|
||
"accuracy": 0.92,
|
||
"detection_classes": json.dumps(["car", "truck", "bus", "motorcycle"]),
|
||
"input_size": "640x640",
|
||
"inference_time": 18.2,
|
||
"is_enabled": True,
|
||
"creator": "admin",
|
||
"tags": json.dumps(["vehicle", "detection"])
|
||
},
|
||
{
|
||
"name": "人脸识别算法",
|
||
"description": "高精度人脸识别算法",
|
||
"version": "1.5.0",
|
||
"model_path": "/models/face_recognition.pt",
|
||
"config_path": "/configs/face.yaml",
|
||
"status": "inactive",
|
||
"accuracy": 0.98,
|
||
"detection_classes": json.dumps(["face"]),
|
||
"input_size": "512x512",
|
||
"inference_time": 25.0,
|
||
"is_enabled": False,
|
||
"creator": "admin",
|
||
"tags": json.dumps(["face", "recognition"])
|
||
}
|
||
]
|
||
|
||
for alg_data in algorithms:
|
||
algorithm = Algorithm(**alg_data)
|
||
db.add(algorithm)
|
||
|
||
db.commit()
|
||
print("✅ 算法数据初始化完成")
|
||
|
||
# 创建示例设备
|
||
devices = [
|
||
{
|
||
"name": "港口A区主监控",
|
||
"device_type": "camera",
|
||
"location": "港口A区",
|
||
"ip_address": "192.168.1.100",
|
||
"port": 554,
|
||
"username": "admin",
|
||
"password": "admin123",
|
||
"rtsp_url": "rtsp://192.168.1.100:554/stream1",
|
||
"status": "online",
|
||
"resolution": "1920x1080",
|
||
"fps": 25,
|
||
"algorithm_id": 1,
|
||
"is_enabled": True,
|
||
"latitude": 22.3193,
|
||
"longitude": 114.1694,
|
||
"description": "港口A区主要监控摄像头",
|
||
"manufacturer": "Hikvision",
|
||
"model": "DS-2CD2T47G1-L",
|
||
"serial_number": "HK123456789"
|
||
},
|
||
{
|
||
"name": "港口B区监控",
|
||
"device_type": "camera",
|
||
"location": "港口B区",
|
||
"ip_address": "192.168.1.101",
|
||
"port": 554,
|
||
"username": "admin",
|
||
"password": "admin123",
|
||
"rtsp_url": "rtsp://192.168.1.101:554/stream1",
|
||
"status": "online",
|
||
"resolution": "1920x1080",
|
||
"fps": 25,
|
||
"algorithm_id": 2,
|
||
"is_enabled": True,
|
||
"latitude": 22.3195,
|
||
"longitude": 114.1696,
|
||
"description": "港口B区监控摄像头",
|
||
"manufacturer": "Dahua",
|
||
"model": "IPC-HFW4431R-ZE",
|
||
"serial_number": "DH987654321"
|
||
},
|
||
{
|
||
"name": "边检站门禁",
|
||
"device_type": "gate",
|
||
"location": "边检站入口",
|
||
"ip_address": "192.168.1.102",
|
||
"port": 80,
|
||
"username": "admin",
|
||
"password": "admin123",
|
||
"status": "online",
|
||
"is_enabled": True,
|
||
"latitude": 22.3190,
|
||
"longitude": 114.1690,
|
||
"description": "边检站入口门禁系统",
|
||
"manufacturer": "Suprema",
|
||
"model": "BioStation 2",
|
||
"serial_number": "SP123456789"
|
||
}
|
||
]
|
||
|
||
for dev_data in devices:
|
||
device = Device(**dev_data)
|
||
db.add(device)
|
||
|
||
db.commit()
|
||
print("✅ 设备数据初始化完成")
|
||
|
||
# 创建示例事件
|
||
events = [
|
||
{
|
||
"event_type": "person_detection",
|
||
"device_id": 1,
|
||
"algorithm_id": 1,
|
||
"severity": "medium",
|
||
"status": "pending",
|
||
"confidence": 0.95,
|
||
"bbox": json.dumps([100, 150, 80, 160]),
|
||
"image_path": "/events/images/person_001.jpg",
|
||
"description": "检测到人员进入监控区域",
|
||
"location": "港口A区",
|
||
"detected_objects": json.dumps([{"type": "person", "confidence": 0.95}]),
|
||
"processing_time": 15.5,
|
||
"is_alert": True,
|
||
"alert_sent": True
|
||
},
|
||
{
|
||
"event_type": "vehicle_detection",
|
||
"device_id": 2,
|
||
"algorithm_id": 2,
|
||
"severity": "low",
|
||
"status": "resolved",
|
||
"confidence": 0.92,
|
||
"bbox": json.dumps([200, 100, 120, 80]),
|
||
"image_path": "/events/images/vehicle_001.jpg",
|
||
"description": "检测到车辆通过",
|
||
"location": "港口B区",
|
||
"detected_objects": json.dumps([{"type": "car", "confidence": 0.92}]),
|
||
"processing_time": 18.2,
|
||
"is_alert": False,
|
||
"alert_sent": False,
|
||
"operator_id": 1,
|
||
"resolution_notes": "正常车辆通行",
|
||
"resolved_at": datetime.utcnow() - timedelta(hours=2)
|
||
},
|
||
{
|
||
"event_type": "intrusion",
|
||
"device_id": 1,
|
||
"algorithm_id": 1,
|
||
"severity": "high",
|
||
"status": "processing",
|
||
"confidence": 0.88,
|
||
"bbox": json.dumps([300, 200, 60, 120]),
|
||
"image_path": "/events/images/intrusion_001.jpg",
|
||
"description": "检测到可疑人员入侵",
|
||
"location": "港口A区",
|
||
"detected_objects": json.dumps([{"type": "person", "confidence": 0.88}]),
|
||
"processing_time": 16.0,
|
||
"is_alert": True,
|
||
"alert_sent": True
|
||
}
|
||
]
|
||
|
||
for evt_data in events:
|
||
event = Event(**evt_data)
|
||
db.add(event)
|
||
|
||
db.commit()
|
||
print("✅ 事件数据初始化完成")
|
||
|
||
print("\n🎉 所有示例数据初始化完成！")
|
||
|
||
except Exception as e:
|
||
print(f"❌ 初始化数据时出错: {e}")
|
||
db.rollback()
|
||
finally:
|
||
db.close()
|
||
|
||
if __name__ == "__main__":
|
||
init_sample_data() |