30 Plant Types Image Detection

由 dima806 开发

这是一个基于Vision Transformer (ViT)架构的图像分类模型，能够准确识别30种不同类型的植物，平均准确率达到93%。

图像分类

Transformers

开源协议:Apache-2.0 #高精度植物识别 #农业图像分类 #ViT视觉模型

下载量 27

发布时间 : 10/29/2023

模型介绍

内容详情

替代品

模型简介

该模型使用google/vit-base-patch16-224-in21k作为基础模型，专门用于植物类型识别任务。它能够从输入图像中预测植物的具体种类，在农业、植物学研究等领域有广泛应用价值。

模型特点

高准确率

在30类植物分类任务上达到93%的整体准确率，部分类别如椰子、红薯等达到99%以上的准确率。

多类别识别

能够识别30种不同的植物类型，涵盖常见农作物、水果和药用植物等。

基于ViT架构

采用先进的Vision Transformer架构，相比传统CNN模型在图像分类任务上有更好的表现。

模型能力

植物种类识别

图像分类

农业图像分析

使用案例

农业

农作物自动识别

用于农田监测系统中自动识别不同作物类型

可准确识别水稻、玉米、木薯等主要农作物

水果品质检测

在水果分拣线上识别不同水果种类

对香蕉、橙子、芒果等水果识别准确率达95%以上

植物学研究

植物标本分类

辅助植物学家快速分类植物标本

可识别包括药用植物如芦荟、姜黄等在内的多种植物

许可证：apache-2.0
评估指标：

准确率
F1分数

基础模型：

google/vit-base-patch16-224-in21k

根据图像预测植物类型，准确率约为93%。

更多详情请参见：https://www.kaggle.com/code/dima806/30-plant-types-image-detection-vit

分类报告：  

              精确率    召回率    F1分数    支持数  

        番石榴     0.9846    0.9600    0.9722       200  
      高良姜     0.9418    0.8900    0.9152       200  
      木苹果     0.9949    0.9750    0.9848       200  
        水稻     0.9731    0.9050    0.9378       200  
        茄子     0.9848    0.9700    0.9773       200  
        黄瓜     0.9561    0.9800    0.9679       200  
      木薯     0.9899    0.9800    0.9849       200  
       木瓜     0.9851    0.9950    0.9900       200  
       香蕉     0.9950    0.9900    0.9925       200  
       橙子     0.9534    0.9200    0.9364       200  
   哈密瓜     0.5271    0.3400    0.4134       200  
      椰子     0.9950    1.0000    0.9975       200  
     大豆     0.9754    0.9900    0.9826       200  
       柚子     0.9563    0.9850    0.9704       200  
    菠萝     0.9703    0.9800    0.9751       200  
        甜瓜     0.5000    0.6150    0.5516       200  
      红葱头     0.9949    0.9750    0.9848       200  
   辣椒     0.9755    0.9950    0.9851       200  
      菠菜     0.9231    0.9600    0.9412       200  
      烟草     0.9151    0.9700    0.9417       200  
     芦荟     0.9949    0.9800    0.9874       200  
      姜黄     0.9005    0.8600    0.8798       200  
       玉米     0.9610    0.9850    0.9728       200  
       生姜     0.8551    0.8850    0.8698       200  
   红薯     1.0000    0.9950    0.9975       200  
       羽衣甘蓝     0.9268    0.9500    0.9383       200  
    长豆角     0.9850    0.9850    0.9850       200  
   西瓜     0.9252    0.9900    0.9565       200  
       芒果     0.9239    0.9100    0.9169       200  
   水苹果     0.8807    0.9600    0.9187       200  

     准确率                         0.9292      6000  
    宏平均     0.9282    0.9292    0.9275      6000  
加权平均     0.9282    0.9292    0.9275      6000