基于飞浆paddle的Mv3驾驶员行为识别

"其实一开始并没有想学深度学习"


bash 复制代码
! pip install --upgrade pip
! pip install paddlex
! pip install --user --upgrade pyarrow==11.0.0
# 配置环境

train_list格式(test同理):图片路径+\t+标签

newLabels格式:标签

训练代码

python 复制代码
import paddlex as pdx

from paddlex import transforms as T

train_transforms = T.Compose(
    [T.RandomCrop(crop_size=224), T.RandomHorizontalFlip(), T.Normalize()])

eval_transforms = T.Compose([
    T.ResizeByShort(short_size=256), T.CenterCrop(crop_size=224), T.Normalize()
])
# 定义数据集的transform

train_dataset = pdx.datasets.ImageNet(
    data_dir='train',
    file_list='train_list.txt',
    label_list='newLabels.txt',
    transforms=train_transforms,
    shuffle=True)
    
eval_dataset = pdx.datasets.ImageNet(
    data_dir='train',
    file_list='val_list.txt',
    label_list='newLabels.txt',
    transforms=eval_transforms)
# 定义数据集

num_classes = len(train_dataset.labels)
model = pdx.cls.MobileNetV3_large_ssld(num_classes=num_classes)
model.train(num_epochs=6, # 训练轮次
            train_dataset=train_dataset, #训练集
            train_batch_size=32,# 训练batch
            eval_dataset=eval_dataset, #测试集
            lr_decay_epochs=[2, 4],# 学习率变化轮次
            save_interval_epochs=2, # 保存模型轮次
            learning_rate=0.00125,# 起始学习率
            save_dir='output/mobilenetv3_large_ssld3',# 保存模型目录
            use_vdl=True)
# 开始训练
相关推荐
chnming19875 分钟前
STL关联式容器之map
开发语言·c++
进击的六角龙7 分钟前
深入浅出:使用Python调用API实现智能天气预报
开发语言·python
檀越剑指大厂7 分钟前
【Python系列】浅析 Python 中的字典更新与应用场景
开发语言·python
罗小罗同学15 分钟前
医工交叉入门书籍分享:Transformer模型在机器学习领域的应用|个人观点·24-11-22
深度学习·机器学习·transformer
湫ccc15 分钟前
Python简介以及解释器安装(保姆级教学)
开发语言·python
孤独且没人爱的纸鹤18 分钟前
【深度学习】:从人工神经网络的基础原理到循环神经网络的先进技术,跨越智能算法的关键发展阶段及其未来趋势,探索技术进步与应用挑战
人工智能·python·深度学习·机器学习·ai
程序伍六七19 分钟前
day16
开发语言·c++
阿_旭21 分钟前
TensorFlow构建CNN卷积神经网络模型的基本步骤:数据处理、模型构建、模型训练
人工智能·深度学习·cnn·tensorflow
羊小猪~~22 分钟前
tensorflow案例7--数据增强与测试集, 训练集, 验证集的构建
人工智能·python·深度学习·机器学习·cnn·tensorflow·neo4j
wkj00123 分钟前
php操作redis
开发语言·redis·php