当前位置: 首页 > news >正文

对于多分类问题,使用深度学习(Keras)进行迁移学习提升性能

本文是仿照前面的文章,使用Keras迁移学习提升性能,原文是针对二分类问题,使用迁移学习的方式来提升准确率,本文用迁移学习的方式来提升多分类问题的准确率。

同时,在前面的文章中,使用普通的小型3层卷积网络+2层全连接层实现了多分类的85%左右的准确率, 此处将用迁移学习的方式来进一步提升性能。

1. 准备数据集

为了演示,本次选用了博文keras系列︱图像多分类训练与利用bottleneck features进行微调(三)中提到的数据集,原始的数据集将所有类别的train照片放到train文件夹中,所有的test照片放在test文件夹中,而用不同数字开头来表示不同类别,比如以3开头的照片就是bus类等。首先将这些不同类别的照片放在不同的文件夹中,最终的train文件夹有5个子文件夹,每个子文件夹中有80张图片,最终的test文件夹中有5个子文件夹,每个子文件夹中有20张图片。总共只有500张图片。

在代码上,需要用ImageDataGenerator来做数据增强,并且用flow_from_directory来从文件夹中产生数据流。
代码和二分类的文章基本相同,唯一的不同之处是要设置class_mode='categorical',而不是原来二分类问题的class_mode='binary'

http://www.lryc.cn/news/208555.html

相关文章:

  • Python----break关键字对while...else结构的影响
  • js实现将文本生成二维码(腾讯云cos)
  • 机架式服务器介绍
  • 解决github有时能访问有时不能访问的问题2
  • Go实现网络通信
  • 在antd里面渲染MarkDown并且自定义一个锚点目录TOC(重点解决导航目录不跟随文档滚动的问题)
  • Linux MMC子系统 - 2.eMMC 5.1总线协议浅析
  • 时序预测 | Python实现ARIMA-LSTM自回归移动差分模型结合长短期记忆神经网络时间序列预测
  • 【Linux】部署单机OA项目及搭建spa前后端分离项目
  • 2023中国计算机大会:蚂蚁集团连发两支百万级科研基金
  • Knife4j使用教程(三) -- 实体类的配置注解(@ApiModel与@ApiModelProperty 的 认识与使用)
  • 计算机网络【CN】IPV4报文格式
  • SQL server数据库单用户模式如何退出
  • QT mqtt 在子线程中使用
  • Tomcat运维以及优化
  • C++设计模式_14_Facade门面模式
  • 正点原子嵌入式linux驱动开发——外置RTC芯片PCF8563
  • 自动驾驶感知算法面经(20+)
  • 计算机操作系统重点概念整理-第二章 进程管理【期末复习|考研复习】
  • 如何学好C++?学习C和C++的技巧是什么?
  • leetcode_39 组合总和
  • 【登录安全测试】
  • MySQL -- 数据类型
  • Windows11搭建kafka-python环境
  • 一篇了解springboot3请求参数种类及接口测试
  • CFD模拟仿真理论知识:流体仿真应用
  • 【springboot单元测试,集成测试】
  • 【音视频】音视频开发与学习
  • 软考系统架构之案例篇(架构设计相关概念)
  • 基于大数据的时间序列股价预测分析与可视化 - lstm 计算机竞赛