Python基于 ImageAI 模块实践 idenprof数据集识别预测分析

最新推荐文章于 2024-08-16 08:21:00 发布

程序员lamed

最新推荐文章于 2024-08-16 08:21:00 发布

阅读量409

点赞数

文章标签： Python 程序员编程

本文链接：https://blog.csdn.net/weixin_45342712/article/details/96986444

版权

本文介绍了使用Python的ImageAI库进行图像识别的实践过程。通过idenprof数据集，作者展示了如何利用预训练模型进行图像分类，并且在测试中取得了55.64%以上的准确率。ImageAI简化了图像识别任务，但也限制了模型的灵活性，它更像是一个开发者社区，可以分享数据集和预训练模型。

摘要由CSDN通过智能技术生成

图像识别早已不是很新鲜的话题了，很多数据处理的任务到最后都会归为图像识别中，在之前的很多工作中，我陆陆续续也接触了很多相关的工作，从最开始数据处理，到模型搭建与最终上线也都经历，大多数时候模型都是自己搭建的，虽然说现在keras的出现极大地简化了模型的搭建工作，但是整个过程还是需要自己去实践完成的，对于很多的初学者来说并不是很容易的。

今天发现了一个好玩的库——ImageAI，简单的说一下我的理解就是对keras的又一层封装，但又不全是这样。ImageAI简化了整个图像识别和目标检测的工作，今天想来简单看看，整个模块的能力。

使用的使用网上公开的数据集 idenprof ，分为train和test两个数据集，每个集合里面共有10个类别，数据集截图如下：
在这里插入图片描述
具体实践如下：

    #!usr/bin/env python
    # encoding:utf-8
    from __future__ import division
     
    """
    __Author__:沂水寒城
    功能： python基于 ImageAI 模块实现 idenprof 数据集预测识别
    当前官网中 ImageAI 主要提供四种类型的预测模型分别如下：
    SqueezeNet（预测速度最快 正确率中等）
    ResNet50 （预测速度快 正确率较高）
    InceptionV3（预测速度慢 正确率高）
    DenseNet121（预测速度更慢 正确率最高）
    """
     
                    我还整理了更多有关Python的免费学习资料和心得，
                           
                            在这里不好一一放出   
                              
                                进群688244617
                              
                                  免费自取 


    import os
    import threading
    from imageai.Prediction import ImagePrediction
    from imageai.Prediction.Custom import ModelTraining
    from imageai