V

Vit Base Patch16 224 In21k Wr

由Imene開發
該模型是基於google/vit-base-patch16-224-in21k在未知數據集上微調的視覺Transformer模型,主要用於圖像分類任務。
下載量 21
發布時間 : 9/7/2022

模型概述

這是一個基於Vision Transformer架構的圖像分類模型,在未知數據集上進行了微調,適用於通用的圖像識別任務。

模型特點

基於預訓練模型微調
基於google/vit-base-patch16-224-in21k預訓練模型進行微調,繼承了強大的圖像特徵提取能力
混合精度訓練
使用mixed_float16精度進行訓練,兼顧訓練速度和模型精度
優化器配置
採用AdamWeightDecay優化器配合PolynomialDecay學習率調度,有助於穩定訓練過程

模型能力

圖像分類
特徵提取

使用案例

計算機視覺
通用圖像分類
可用於對常見物體和場景進行分類識別
驗證準確率57.7%,前三準確率80.35%
AIbase
智啟未來,您的人工智能解決方案智庫
© 2025AIbase