基于vit-base模型的植物分类模型,使用kaggle的Plant dataset进行训练,该数据集包括30个种类的30000张图片。
数据集Plants Type Dataset
微调基础模型vit-base日常物品分类
git clone https://www.modelscope.cn/reymond/vit_base_plants_classification.git
git clone https://www.modelscope.cn/reymond/vit_base_plants_classification.git
评论