• 对于多分类问题,使用深度学习(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'
    
    • 1
    • 2

  • 相关阅读:
    yolov2
    3DMax
    Nacos服务心跳和健康检查源码介绍
    c语言之二级指针
    python工作中、使用中必须会的使用方法
    2024 ,Android 15 预览版来了
    odoo javascript参考(七)
    自动化测试也可以不写代码?今天就教你
    多线程 CompletableFuture(1)
    陇剑杯线上——机密内存取证
  • 原文地址:https://blog.csdn.net/weixin_44119674/article/details/134049451