Driving-Behavior-Recognition

训练驾驶行为识别模型

环境

python 3.7
torch 1.7
torchvision 0.8
onnxruntime 1.7
opencv-python 4.5

该项目包含训练驾驶行为识别模型的所有相关处理代码。包括数据集处理、模型训练、模型转换、模型测试等

使用convertVideoToImages.py将采集视频转化为图片
使用打标签工具actionlabel进行数据标注
使用ceateTxt.py创建数据集标签文件

train.py是模型训练代码，使用时注意修改相关参数
vlr40_train.py是精简版训练代码,可以使用VLR40数据集训练车标识别模型
VLR40数据集仅供学习使用，链接：https://pan.baidu.com/s/1SctXvwqWe5WlfkwcmrPXQw 提取码：75c3

torchToOnnx.py将pth格式权重转化为通用onnx格式权重

predict.py使用模型进行检测

Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
VLR40/txt		VLR40/txt
weights		weights
.gitignore		.gitignore
CAM.py		CAM.py
Fcalayer.py		Fcalayer.py
Grad_CAM.py		Grad_CAM.py
README.md		README.md
classification_sample.py		classification_sample.py
convertImage.py		convertImage.py
convertVideo.py		convertVideo.py
convertVideoToImages.py		convertVideoToImages.py
createTxt.py		createTxt.py
detect.py		detect.py
detect_cap_multi.py		detect_cap_multi.py
detect_multi.py		detect_multi.py
efficientnet.py		efficientnet.py
ffmpeg_push.py		ffmpeg_push.py
ghost_net.py		ghost_net.py
ghostnet.py		ghostnet.py
grad_cam_pytorch.py		grad_cam_pytorch.py
kd.py		kd.py
mnext.py		mnext.py
mobilenet.py		mobilenet.py
mobilenetv2_cam.py		mobilenetv2_cam.py
mobilenetv2_cbam.py		mobilenetv2_cbam.py
mobilenetv3.py		mobilenetv3.py
mobilenetv3_2.py		mobilenetv3_2.py
mobilenetv3_torch.py		mobilenetv3_torch.py
model_test.py		model_test.py
multiprocess.py		multiprocess.py
open_rtmp.py		open_rtmp.py
predict.py		predict.py
read_csv.py		read_csv.py
request.py		request.py
server.py		server.py
show_lr.py		show_lr.py
split_dataset.py		split_dataset.py
t_tqdm.py		t_tqdm.py
torchToOnnx.py		torchToOnnx.py
train.py		train.py
transforms.py		transforms.py
utils.py		utils.py
vlr40_train.py		vlr40_train.py
voc2classify.py		voc2classify.py