基于飞浆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)
# 开始训练
相关推荐
秋の花6 分钟前
【JAVA基础】Java集合基础
java·开发语言·windows
小松学前端9 分钟前
第六章 7.0 LinkList
java·开发语言·网络
charles_vaez14 分钟前
开源模型应用落地-glm模型小试-glm-4-9b-chat-快速体验(一)
深度学习·语言模型·自然语言处理
可峰科技18 分钟前
斗破QT编程入门系列之二:认识Qt:编写一个HelloWorld程序(四星斗师)
开发语言·qt
全栈开发圈22 分钟前
新书速览|Java网络爬虫精解与实践
java·开发语言·爬虫
面试鸭27 分钟前
离谱!买个人信息买到网安公司头上???
java·开发语言·职场和发展
小白学大数据27 分钟前
JavaScript重定向对网络爬虫的影响及处理
开发语言·javascript·数据库·爬虫
Python大数据分析@31 分钟前
python操作CSV和excel,如何来做?
开发语言·python·excel
黑叶白树32 分钟前
简单的签到程序 python笔记
笔记·python
YRr YRr36 分钟前
深度学习:Transformer Decoder详解
人工智能·深度学习·transformer