imgclsmob

Глубокие сети обучения

Это репо используется для исследования сверточных сетей, прежде всего, для задач компьютерного зрения. Для этой цели репо содержит (Re) реализации различных моделей классификации, сегментации, обнаружения и позы и сценариев для обучения/оценки/преобразования.

Используются следующие рамки:

• Mxnet/gluon (info),
• Pytorch (info),
• Цепь (информация),
• Керас (информация),
• Tensorflow 1.x/2.x (информация).

Для каждой поддерживаемой структуры есть PIP-Package, содержащий чистые модели без вспомогательных сценариев. Список пакетов:

• Gluoncv2 для Gluon,
• pytorchcv для pytorch,
• chailercv2 для цепи,
• kerascv для керас,
• tensorflowcv для tensorflow 1.x,
• TF2CV для TensorFlow 2.x.

В настоящее время модели в основном реализованы на Gluon, а затем переносятся в другие рамки. Некоторые модели предварительно предварительно на ImageNet-1K, Cifar-10/100, SVHN, Cub-200-2011, Pascal VOC2012, ADE20K, CityScapes и Coco Dataes. Все предварительные веса загружаются автоматически во время использования. См. Примеры такой автоматической загрузки весов в соответствующих разделах документации, посвященной конкретному пакету:

• Gluon Models,
• Модели Pytorch,
• Цепные модели,
• Керас модели,
• Tensorflow 1.x модели,
• Tensorflow 2.x модели.

Установка

Чтобы использовать обучение/оценку сценариев, а также все модели, вам необходимо клонировать репозиторий и установить зависимости:

 git clone [email protected]:osmr/imgclsmob.git
pip install -r requirements.txt

Таблица реализованных моделей классификации

Некоторые замечания:

• Repo - это репозиторий автора, если он существует.
• a , b , c , d и e означает реализацию модели для ImageNet-1K, CiFAR-10, CIFAR-100, SVHN и CUB-200111, соответственно.
• A , B , C , D и E означает наличие предварительно обученной модели для соответствующих наборов данных.

Модель	Глюон	Пирог	Цепь	Керас	Тфу	TF2	Бумага	Репо	Год
Алекснет	А	А	А	А	А	А	связь	связь	2012
Zfnet	А	А	А	А	А	А	связь	-	2013
Vgg	А	А	А	А	А	А	связь	-	2014
BN-VGG	А	А	А	А	А	А	связь	-	2015
BN-Incept	А	А	А	-	-	А	связь	-	2015
Resnet	Abcde	Abcde	Abcde	А	А	Abcde	связь	связь	2015
PRERESNET	ABCD	ABCD	ABCD	А	А	ABCD	связь	связь	2016
Resnext	ABCD	ABCD	ABCD	А	А	ABCD	связь	связь	2016
Сена	А	А	А	А	А	А	связь	связь	2017
SE-Resnet	Abcde	Abcde	Abcde	А	А	Abcde	связь	связь	2017
Se-Preresnet	ABCD	ABCD	ABCD	А	А	ABCD	связь	связь	2017
Se-Resnext	А	А	А	А	А	А	связь	связь	2017
Resnest (а)	А	А	А	-	-	А	связь	связь	2020
IBN-Resnet	А	А	-	-	-	А	связь	связь	2018
IBN-RESNEXT	А	А	-	-	-	А	связь	связь	2018
IBN-DENSENET	А	А	-	-	-	А	связь	связь	2018
Самолет	А	А	А	-	-	А	связь	связь	2018
AirNext	А	А	А	-	-	А	связь	связь	2018
Bam-Resnet	А	А	А	-	-	А	связь	связь	2018
CBAM-RESNET	А	А	А	-	-	А	связь	связь	2018
Resattnet	а	а	а	-	-	-	связь	связь	2017
Скнет	а	а	а	-	-	-	связь	связь	2019
Сокровище	А	А	А	-	-	А	связь	связь	2020
Регнет	А	А	А	-	-	А	связь	связь	2020
Диа-Реснет	ABCD	ABCD	ABCD	-	-	-	связь	связь	2019
DiA-PreresNet	ABCD	ABCD	ABCD	-	-	-	связь	связь	2019
Pyramidnet	ABCD	ABCD	ABCD	-	-	ABCD	связь	связь	2016
Diracnetv2	А	А	А	-	-	А	связь	связь	2017
ShareSnet	а	а	а	-	-	-	связь	связь	2017
Cru-Net	А	-	-	-	-	-	связь	связь	2018
Денсенет	ABCD	ABCD	ABCD	А	А	ABCD	связь	связь	2016
Конденсенет	А	А	А	-	-	-	связь	связь	2017
Sparsenet	а	а	а	-	-	-	связь	связь	2018
Пелинет	А	А	А	-	-	А	связь	связь	2018
Oct-Resnet	ABCD	а	а	-	-	-	связь	-	2019
Res2net	а	-	-	-	-	-	связь	-	2019
WRN	ABCD	ABCD	ABCD	-	-	а	связь	связь	2016
WRN-1BIT	BCD	BCD	BCD	-	-	-	связь	связь	2018
DRN-C	А	А	А	-	-	А	связь	связь	2017
DRN-D	А	А	А	-	-	А	связь	связь	2017
Дпн	А	А	А	-	-	А	связь	связь	2017
Darknet Ref	А	А	А	А	А	А	связь	связь	-
Darknet Tiny	А	А	А	А	А	А	связь	связь	-
Darknet-19	а	а	а	а	а	а	связь	связь	-
Darknet-53	А	А	А	А	А	А	связь	связь	2018
Канал	а	а	а	-	а	-	связь	связь	2018
ISQRT-COV-RESNET	а	а	-	-	-	-	связь	связь	2017
Revnet	-	а	-	-	-	-	связь	связь	2017
I-REVNET	А	А	А	-	-	-	связь	связь	2018
Бакнет	А	А	А	-	-	А	связь	связь	2019
DLA	А	А	А	-	-	А	связь	связь	2017
Msdnet	а	Аб	-	-	-	-	связь	связь	2017
Рыбная сеть	А	А	А	-	-	-	связь	связь	2018
Espnetv2	А	А	А	-	-	-	связь	связь	2018
Dicenet	А	А	А	-	-	А	связь	связь	2019
Hrnet	А	А	А	-	-	А	связь	связь	2019
Vovnet	А	А	А	-	-	А	связь	связь	2019
SELECSLS	А	А	А	-	-	А	связь	связь	2019
Жесткая сетя	А	А	А	-	-	А	связь	связь	2019
X-DENSENET	ABCD	ABCD	ABCD	-	-	-	связь	связь	2017
Squeezenet	А	А	А	А	А	А	связь	связь	2016
Squeezeresnet	А	А	А	А	А	А	связь	-	2016
Squeezenext	А	А	А	А	А	А	связь	связь	2018
Shufflenet	А	А	А	А	А	А	связь	-	2017
Shufflenetv2	А	А	А	А	А	А	связь	-	2018
Мене	А	А	А	А	А	А	связь	связь	2018
Mobilenet	Аэ	Аэ	Аэ	А	А	Аэ	связь	связь	2017
FD-Mobilenet	А	А	А	А	А	А	связь	связь	2018
Mobilenetv2	А	А	А	А	А	А	связь	связь	2018
Mobilenetv3	А	А	А	А	-	А	связь	связь	2019
IGCV3	А	А	А	А	А	А	связь	связь	2018
Призрак	а	а	а	-	-	а	связь	связь	2019
Mnasnet	А	А	А	А	А	А	связь	-	2018
ДАРТС	А	А	А	-	-	-	связь	связь	2018
Proxylessnas	Аэ	Аэ	Аэ	-	-	Аэ	связь	связь	2018
Fbnet-c	А	А	А	-	-	А	связь	-	2018
Xception	А	А	А	-	-	А	связь	связь	2016
Началов3	А	А	А	-	-	А	связь	связь	2015
Началова4	А	А	А	-	-	А	связь	связь	2016
Началоререснев1	А	А	А	-	-	А	связь	связь	2016
Началоререснев2	А	А	А	-	-	А	связь	связь	2016
Полинет	А	А	А	-	-	А	связь	связь	2016
NASNET-LARGE	А	А	А	-	-	А	связь	связь	2017
Nasnet-mobile	А	А	А	-	-	А	связь	связь	2017
Pnasnet-Large	А	А	А	-	-	А	связь	связь	2017
Spnasnet	А	А	А	-	-	А	связь	связь	2019
EfficeNet	А	А	А	А	-	А	связь	связь	2019
Микнет	А	А	А	-	-	А	связь	связь	2019
Нин	BCD	BCD	BCD	-	-	-	связь	связь	2013
ROR-3	BCD	BCD	BCD	-	-	-	связь	-	2016
Рир	BCD	BCD	BCD	-	-	-	связь	-	2016
Resdrop-Resnet	BCD	BCD	BCD	-	-	-	связь	связь	2016
Встряхнуть-рубеж	BCD	BCD	BCD	-	-	-	связь	связь	2017
Shakedrop-Resnet	BCD	BCD	BCD	-	-	-	связь	-	2018
Fractalnet	до н.э.	до н.э.	-	-	-	-	связь	связь	2016
NTS-Net	Эн	Эн	Эн	-	-	-	связь	связь	2018

Таблица реализованных моделей сегментации

Некоторые замечания:

• a/A соответствует Pascal VOC2012.
• b/B соответствует ADE20K.
• c/C соответствует городским пейзажам.
• d/D соответствует Coco.
• e/E соответствует Celebamask-HQ.

Модель	Глюон	Пирог	Цепь	Керас	Тфу	TF2	Бумага	Репо	Год
Pspnet	ABCD	ABCD	ABCD	-	-	ABCD	связь	-	2016
DeepLabv3	ABCD	ABCD	ABCD	-	-	ABCD	связь	-	2017
FCN-8S (D)	ABCD	ABCD	ABCD	-	-	ABCD	связь	-	2014
ICnet	В	В	В	-	-	В	связь	связь	2017
Синет	В	В	В	-	-	в	связь	связь	2019
Бисенет	эн	эн	эн	-	-	эн	связь	-	2018
Данет	В	В	В	-	-	В	связь	связь	2018
Быстрый скрининг	В	В	В	-	-	В	связь	-	2019
CGNET	в	в	в	-	-	в	связь	связь	2018
Dabnet	в	в	в	-	-	в	связь	связь	2019
Fpenet	в	в	в	-	-	в	связь	-	2019
Контекст	-	в	-	-	-	-	связь	-	2018
Леднет	в	в	в	-	-	в	связь	-	2019
Эснет	-	в	-	-	-	-	связь	-	2019
Эданет	-	в	-	-	-	-	связь	связь	2018
Энет	-	в	-	-	-	-	связь	-	2016
Erfnet	-	в	-	-	-	-	связь	-	2017
Linknet	-	в	-	-	-	-	связь	-	2017
Segnet	-	в	-	-	-	-	связь	-	2015
U-Net	-	в	-	-	-	-	связь	-	2015
SQNet	-	в	-	-	-	-	связь	-	2016

Таблица реализованных моделей обнаружения объектов

Некоторые замечания:

• a/A соответствует COCO.

Модель	Глюон	Пирог	Цепь	Керас	Тфу	TF2	Бумага	Репо	Год
Вежливый	а	а	а	-	-	а	связь	связь	2019

Таблица реализованных моделей оценки позы человека

Некоторые замечания:

• a/A соответствует COCO.

Модель	Глюон	Пирог	Цепь	Керас	Тфу	TF2	Бумага	Репо	Год
Альфапоза	А	А	А	-	-	А	связь	связь	2016
Простой	А	А	А	-	-	А	связь	связь	2018
SimplePose (Mobile)	А	А	А	-	-	А	связь	-	2018
Легкий открытый	А	А	А	-	-	А	связь	связь	2018
Ibppose	А	А	А	-	-	А	связь	связь	2019

Таблица реализованных моделей автоматического распознавания речи

Некоторые замечания:

• a/A соответствует Librispeech.
• b/B соответствует голосу Mozilla Common.

Модель	Глюон	Пирог	Цепь	Керас	Тфу	TF2	Бумага	Репо	Год
Джаспер Др	Аб	Аб	Аб	-	-	Аб	связь	связь	2019
QuartzNet	Аб	Аб	Аб	-	-	Аб	связь	связь	2019