CSPNet (CSPResNet, CSPDarkNet, DarkNet)

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

Сети Cross Stage Partial Networks разделяют признаки каждого этапа на две части: одна проходит через блоки этапа, а другая объединяется обратно в конце этапа. Этот модуль содержит CSPResNet, CSPResNeXt, CSPDarkNet, DarkNet и варианты timm cs3*.

Статьи

• CSPNet: новый базовый блок, способный повысить способность CNN к обучению
• YOLOv3: Постепенное улучшение

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

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

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

Поддержка Model API

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

Архитектуры

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

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

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

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

теги

ra-теги

Другое