timm documentation

DenseNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

DenseNet

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

Плотные свёрточные сети, в которых каждый слой в плотном блоке принимает на вход конкатенированные выходы всех предыдущих слоёв блока. Адаптировано из torchvision, densenetblur использует blur pool для антиалиасинга при даунсемплинге.

Статьи

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

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

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

Поддержка Model API

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

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

Архитектуры

АрхитектураПараметры (млн)GMACsРазмер входных данныхПредобученные веса
densenet1217.982.872242
densenet16128.687.792241
densenet16914.153.402241
densenet20120.014.342241
densenet264d72.7413.572240
densenetblur121d8.003.112241

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

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

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

Веса группируются по источнику: статья или происхождение, указанные в конфигурации предобученной модели, иначе организация на Hugging Face, отличная от timm, иначе первая часть тега предобученной модели (architecture.tag). Источники с одним весом перечислены в разделе other.

tv tags

ВесРазмер входных данныхTop-1 (%)Размер выборки для оценкиКлассыЛицензияHub
densenet121.tv_in1k22474.7502241000apache-2.0timm
densenet161.tv_in1k22477.3842241000apache-2.0timm
densenet169.tv_in1k22475.9222241000apache-2.0timm
densenet201.tv_in1k22477.2842241000apache-2.0timm

ra-теги

ВесРазмер входных данныхTop-1 (%)Размер выборки для оценкиКлассыЛицензияHub
densenet121.ra_in1k224 (тест 288)76.4862881000apache-2.0timm
densenetblur121d.ra_in1k224 (тест 288)77.3202881000apache-2.0timm
Update on GitHub