/ Модель

Характеристики

Параметры
113 млн
Лицензия
Apache 2.0
Архитектура
vit · dense
Базовая модель
vit-base-patch16-224
Опубликована
5 авг 2025

UCSCPhytoViT83

Классификация изображенийИзображение → label3 профессии

Модель Vision Transformer для классификации изображений фитопланктона на 83 класса по данным Imaging FlowCytobot.

365скачиваний за 30 дней0лайковвеса обновлены6 авг 2025

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

UCSCPhytoViT83 — модель Vision Transformer для классификации изображений фитопланктона по 83 классам. Её разработал Patrick Daniel, дообучив Google ViT-Base с патчами 16×16 на примерно 558 тысячах размеченных изображений, полученных приборами Imaging FlowCytobot и объединённых из наборов IFCB104, IFCB161 и IFCB116. Модель принимает изображения, предварительно приведённые к размеру 224×224 и нормализованные, и выдаёт класс фитопланктона.

Модель предназначена для анализа изображений IFCB, прежде всего из залива Монтерей и залива Сан-Франциско, и ориентирована на исследователей. В README приведены метрики precision, recall и F1 для классов на отложенной выборке. Доступны веса safetensors и формат ONNX. Лицензия — Apache 2.0.

Качество классификации может снижаться на изображениях с других приборов и регионов. Модель не предназначена для изображений, не относящихся к фитопланктону; частицы и клетки с низким содержанием хлорофилла представлены недостаточно, поскольку обучение использовало IFCB с триггером PMT-B. Предсказания рекомендуется проверять с экспертами.

Сценарии применения

  • классификация видов фитопланктона на изображениях IFCB
  • анализ проб фитопланктона из залива Монтерей
  • автоматическая разметка изображений микроскопических организмов
  • исследование состава сообществ фитопланктона

Ограничения

Модель может плохо обобщаться на данные других приборов и регионов и не предназначена для изображений вне области фитопланктона. Частицы и клетки с низким содержанием хлорофилла представлены недостаточно; предсказания следует проверять с экспертами.

Для каких профессий подходит3

Разделы каталога профессий, в задачах которых используется эта модель

Все профессии

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

Model Card for phytoViT_558k_Aug2025

Model Details

Model Description

UCSCPhytoViT83 is a Vision Transformer (ViT) model fine-tuned for image classification of phytoplankton species using labeled images collected from the Imaging FlowCytobot (IFCB) at UCSC. The model is fine-tuned from the pre-trained google/vit-base-patch16-224-in21k base model. The model was trained on images that are aggregated from IFCB104, IFCB161, and IFCB116

  • Developed by: Patrick Daniel
  • Model type: Vision Transformer for image classification
  • License: Apache 2.0
  • Finetuned from model: google/vit-base-patch16-224-in21k

Model Sources

  • Repository: [More Information Needed]

Uses

Direct Use

This model can be used directly for classifying phytoplankton images captured by Imaging FlowCytobots instruments. Focus has been on capturing the variability of the phytoplankton community in Monterey Bay, CA, USA. It is intended for researchers.

Images should be transformed before inference:

    transforms.Compose([
    transforms.Resize((224, 224)),  # match ViT input size
    transforms.Normalize(mean=(0.485, 0.456, 0.406),
                         std=(0.229, 0.224, 0.225))
])

Out-of-Scope Use

This model is not intended for classifying non-phytoplankton images or images from different microscopy systems without proper retraining or adaptation. For the IFCB, the model was trained for instruments that are triggering on PMT-B, so particles and cells with no or limited chlorophyll may not be well represented here.

Bias, Risks, and Limitations

The model was trained on IFCB images collected at UCSC/MBARI and mostly in Monterey Bay, CA, USA or San Francisco Bay, CA, USA and may not generalize well to images from other instruments or regions. Users should validate model predictions with domain experts when possible.

How to Get Started with the Model

Install the transformers library and load the model as shown in the example above. For best results, use images preprocessed similarly to the IFCB dataset (see above).

Training Details

Training Data

The model was trained on approximately 558,000 labeled IFCB images representing 83 classes.

Training Procedure

  • Preprocessing: Images were resized and normalized consistent with ViT base requirements.

Evaluation

Testing Data, Factors & Metrics

  • The model was evaluated on a held-out test set of IFCB images.
  • Metrics include accuracy, precision, recall, and F1-score across phytoplankton classes.

Results

Label Nameprecisionrecallf1-scoreEval #
Akashiwo0.9804050.9846560.9825262998
Alexandrium0.9723280.9686420.9704822902
Amylax_Gonyaulax_Protoceratium0.9872340.9830510.985138236
Asterionellopsis0.9828770.9795220.981197586
Asteromphalus0.9904880.9881380.989311843
Bad_setae0.9815810.9696740.9755911319
Centric0.8861330.8487790.8670542989
Ceratium_divaricatum0.9948250.970960.982748792
Ceratium_furca0.9622020.9661720.9641831212
Ceratium_lineatum0.9759920.9862870.9811121896
Chaetoceros0.9445370.9480.9462653000
Ciliate_large0.9583330.9745760.966387118
Ciliate_large_20.9590910.967890.96347218
Ciliate_other_morpho_10.9155780.9183470.91696992
Clusterflagellate_morpho_10.9945390.9824680.9884671483
Clusterflagellate_morpho_20.9927340.9963540.994541097
Corethron0.9988890.997780.998334901
Cryptophyte0.9519770.9683910.9601141740
Cylindrotheca0.9252590.9691430.9466931750
Detonula_Cerataulina_Lauderia0.8408660.8806670.8603063000
Detritus0.9719750.9879150.979882317
Detritus_infection0.9967170.9963080.9965132438
Dictyocha0.9977050.9954210.9965622184
Dinoflagellate_cyst11117
Dinoflagellate_morpho_10.950980.9847720.967581394
Dinoflagellate_morpho_20.932530.9400810.936292470
Dinophysis0.9869710.9885810.9877751226
Ditylum0.9946190.9964060.9955121113
Entomoneis0.9726260.9784850.9755471162
Eucampia0.9771530.9266670.951243000
Euglenoid0.9724080.9651450.9687632410
Flagellate_morpho_10.9661530.961320.9637312999
Flagellate_morpho_20.9422110.9740260.957854385
Flagellate_morpho_30.9512590.9693330.9602113000
Flagellate_nano_10.9568180.9813520.96893429
Flagellate_nano_20.9881240.9788240.983452425
Fragilariopsis0.9000640.9396670.9194393000
Guinardia_Dactyliosolen0.8068180.9136030.856897544
Gymnodinium0.8307480.8674520.848703679
Gyrodinium0.9886040.9914290.9900141050
Gyrosigma0.9462370.9462370.94623793
Haptophyte_prymnesium0.6226420.6734690.64705949
Hemiaulus0.9032260.9032260.903226155
Hemiselmis0.9508620.9740.9622923000
Heterocapsa_long0.9587630.8942310.925373104
Heterocapsa_rotundata0.9645090.8842110.9226161045
Heterocapsa_triquetra0.8035710.6569340.722892137
Heterosigma_akashiwo10.9984770.9992381313
Laboea0.9905210.9874020.988959635
Leptocylindrus0.9655580.9497660.957597856
Margalefidinium0.9731410.9753780.9742583046
Mesodinium0.95830.9629330.9606112482
Nano_cluster0.9829550.9971180.989986347
Nano_p_white0.9822980.9759510.9791142786
Noctiluca10.9655170.98245629
Odontella11130
Pennate0.9093320.8646950.8864523178
Pennate_Tropidoneis0.8372090.7422680.78688597
Pennate_Unknown0.841270.8281250.83464664
Pennate_small0.8433730.8641980.853659405
Peridinium0.9684350.9690860.968761488
Phaeocystis0.9945020.9979310.9962131450
Pleurosigma0.9913790.9631490.97706597
Polykrikos0.9970990.9951740.9961351036
Proboscia0.9925930.9852940.98893136
Prorocentrum_narrow0.9819520.9819520.9819522992
Prorocentrum_wide0.9888930.9914630.9901762694
Pseudo-nitzschia0.9563240.9776740.9668811075
Pyramimonas10.9823790.991111227
Rhizosolenia0.9960080.9842210.990079507
Scrippsiella0.9605880.9310150.945571754
Skeletonema0.986320.9931130.9897051452
Spiky_pacman0.9610720.9589080.9599893553
Stombidinium_morpho_10.9198470.9094340.914611265
Strombidinum_morpho_20.9663990.9406330.9533422813
Thalassionema0.9898820.9915540.990717592
Thalassiosira0.9242720.9316670.9279553000
Tiarina0.9978430.9967670.997305928
Tontonia0.9541670.9385250.946281244
Torodinium0.9947920.9904930.9926381157
Tropidoneis10.9935690.996774311
Vicicitus0.9432840.9546830.948949331
haptophyte_ucynA_host10.9985320.9992652043
accuracy0.9586620.9586620.9586620.958662
macro avg0.9539730.9516580.952527111810
weighted avg0.9589480.9586620.958652111810

Confusion Matrix

Technical Specifications

Model Architecture and Objective

Citation

If you use this model in your research, please cite:

APA:

Daniel, P. (2025). phytoViT_558k_Aug2025: Vision Transformer model for phytoplankton image classification. Retrieved from https://huggingface.co/phytoViT_558k_Aug2025

BibTeX:

@misc{daniel2025phytoViT,
  author = {Patrick Daniel},
  title = {phytoViT_558k_Aug2025: Vision Transformer model for phytoplankton image classification},
  year = {2025},
  howpublished = {\url{https://huggingface.co/phytoViT_558k_Aug2025}},
}

Model Card Authors

Patrick Daniel

Model Card Contact

pcdaniel@ucsc.edu

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм · image-classification

Можно запустить

Подбираем GPU…

Класс архитектуры
ViTForImageClassification
Ревизия весов
986b17c

0 моделей