8.5 基于深度学习的疲劳驾驶和盲点检测系统
本项目是一个基于PyTorch的单发多框检测器(SSD)训练框架,支持在Pascal VOC和Open Images数据集上进行训练。提供了灵活的网络架构选择,包括VGG16、MobileNetV1、MobileNetV1 Lite、SqueezeNet Lite和MobileNetV2 Lite。用户可通过命令行参数配置数据集类型、网络架构、学习率等训练参数,并支持模型的断点续训。项目实现了基础训练和验证功能,可用于目标检测任务和盲点检测任务。最终在Jetson Nano设备中实现了一个双摄像头系统,通过两个并行的视频流进行实时的疲劳驾驶检测和盲点检测功能。
实例5-8:疲劳驾驶和盲点检测系统(codes/5/drowsiness-and-blind-spot-detection)
8.5.1 技术栈
- PyTorch:作为主要的