Research Article

Automatic Mushroom Species Classification Model for Foodborne Disease Prevention Based on Vision Transformer

Table 2

ViT-L/32 vs. SOTA CNN.

ModelTypesEvaluation metrics (%)
PrecisionSensitivityF1-scoreACCAUC

VGG-16Macro average81.8579.7180.2581.3192.95
Weighted average81.5981.3181.07

ResNet-34Macro average85.6484.7284.8885.7994.43
Weighted average85.8585.7985.59

Inception-V3Macro average82.2181.9181.9483.0492.88
Weighted average83.0183.0482.90

Inception-ResNet-V2Macro average85.1384.2984.5485.3994.42
Weighted average85.4485.3985.25

XceptionMacro average92.8592.6192.6492.9597.82
Weighted average93.2092.9592.98

ViT-L/32Macro average95.6995.9295.7795.9799.01
Weighted average96.0595.9795.99