timm documentation

DLA

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

DLA

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

Сети Deep Layer Aggregation (DLA) объединяют признаки между блоками и стадиями с помощью итеративных и иерархических узлов агрегации. Включает варианты DLA, построенные на основе блоков Res2Net.

Статьи

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

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

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

Поддержка Model API

ВозможностьDLA
карты признаков 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()✓

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

Архитектуры

АрхитектураПараметры (млн)GMACsРазмер входных данныхПредобученные веса
dla3415.743.072241
dla46_c1.300.582241
dla46x_c1.070.542241
dla6022.044.262241
dla60_res2net20.854.152241
dla60_res2next17.033.492241
dla60x17.353.542241
dla60x_c1.320.592241
dla10233.277.192241
dla102x26.315.892241
dla102x241.289.342241
dla16953.3911.602241

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

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

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

ВесРазмер входных данныхTop-1 (%)Размер выборки для оценкиКлассыЛицензияHub
dla34.in1k22474.6322241000bsd-3-clausetimm
dla46_c.in1k22464.8802241000bsd-3-clausetimm
dla46x_c.in1k22466.0082241000bsd-3-clausetimm
dla60.in1k22477.0362241000bsd-3-clausetimm
dla60_res2net.in1k22478.4802241000unknowntimm
dla60_res2next.in1k22478.4882241000unknowntimm
dla60x.in1k22478.2342241000bsd-3-clausetimm
dla60x_c.in1k22467.9282241000bsd-3-clausetimm
dla102.in1k22478.0162241000bsd-3-clausetimm
dla102x.in1k22478.5162241000bsd-3-clausetimm
dla102x2.in1k22479.4482241000bsd-3-clausetimm
dla169.in1k22478.7022241000bsd-3-clausetimm
Update on GitHub