timm documentation

InceptionNeXt

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

InceptionNeXt

Категория: CNN · Код: timm/models/inception_next.py · Архитектуры: 4 · Предобученные веса: 5 (5 с результатами на ImageNet-1k)

InceptionNeXt декомпозирует глубокую свёртку с большим ядром из ConvNeXt на параллельные ветви: маленькое квадратное ядро, два ортогональных полосных ядра и тождественное отображение, следуя подходу Inception.

Статьи

Использование

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

>>> import timm
>>> model = timm.create_model('inception_next_tiny.sail_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Поддержка Model API

ВозможностьMetaNeXt
карты признаков features_only=True✓ (шаги 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
Любой размер входных данных, изменения не требуются✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
Послойное затухание LR (group_matcher())✓
reset_classifier()✓

✓ поддерживается, ✓ (квадрат) работает только для квадратных размеров, ✗ не поддерживается, – не требуется (модель уже принимает входные данные любого размера). Проверено путём запуска каждой функции на случайно инициализированной модели (MetaNeXt с inception_next_atto при 224), ожидается, что другие архитектуры того же класса будут вести себя аналогично. Подробности см. на странице Model API.

Архитектуры

АрхитектураПараметры (млн)GMACsРазмер входных данныхПредобученные веса
inception_next_atto4.160.502241
inception_next_base86.6714.852242
inception_next_small49.378.362241
inception_next_tiny28.064.192241

GMACs и количество параметров указаны для размера входных данных и классификатора по умолчанию для каждой архитектуры, согласно результатам бенчмарков.

Предобученные веса

Top-1 — это точность на валидационном наборе ImageNet-1k из results-imagenet.csv для указанного размера оценки, - если для весов нет результата ImageNet-1k (например, веса для предобучения или изображения-текст). Classes — размер предобученного классификатора, none для весов без классификатора.

ВесРазмер входных данныхTop-1 (%)Размер выборки для оценкиКлассыЛицензияHub
inception_next_atto.sail_in1k22475.3482241000apache-2.0timm
inception_next_base.sail_in1k22484.0982241000apache-2.0timm
inception_next_base.sail_in1k_38438485.2023841000apache-2.0timm
inception_next_small.sail_in1k22483.5842241000apache-2.0timm
inception_next_tiny.sail_in1k22482.5082241000apache-2.0timm
Update on GitHub