这是一个人脸追踪风扇的PC控制部分。实现了人脸识别并追踪。
下载模型权重文件后
model_dir='best.pt' 标明权重文件位置
ser_send = False Tcp_send = True 选择控制命令传送方式(双false关闭控制功能)
运行infer.py脚本,会将识别结果和追踪点标出。
PC控制器脚本:infer.py
此脚本实现实时图像采集、目标识别、基于识别结果计算控制变量、将控制目标展示与传输给风扇执行。 前两大功能基于原有脚本detect_face.py修改而来。后两个大功能基于mcu_lab中的多个库实现。包括控制算法my_control.py、TCP和串口通信方式的实现My_serial.py,my_tcpServer.py、离散控制所需要的计时功能stopWatch.py。
其中my_control.py中定义的两个滤波器使得对目标的追踪可以十分稳定。
注:使用onnx模型推理需要onnxruntime python库
风扇执行器系统设计: https://github.com/gb16001/STM32_Pan-tilt_fan
fork from: yolov5face
dataset:wider face