Dieses Repo wird verwendet, um Faltungsnetzwerke hauptsächlich für Computer -Vision -Aufgaben zu erforschen. Zu diesem Zweck enthält das Repo (RE) Implementierungen verschiedener Klassifizierungs-, Segmentierungs-, Erkennungs- und Pose -Schätzmodelle und -Skripte für das Training/Evaluieren/Konvertieren.
Die folgenden Frameworks werden verwendet:
Für jedes unterstützte Framework gibt es ein Pip-Package mit reinen Modellen ohne Hilfskripte. Liste der Pakete:
Derzeit werden Modelle hauptsächlich auf Gluon implementiert und dann auf andere Frameworks portiert. Einige Modelle sind auf ImageNET-1K, CIFAR-10/100, SVHN, CUB-200-2011, PASCAL VOC2012, ADE20K, Cityscapes und Coco-Datensätzen vorgebracht. Alle vorbereiteten Gewichte werden während der Verwendung automatisch geladen. In den entsprechenden Abschnitten der Dokumentation finden Sie Beispiele für eine solche automatische Belastung von Gewichten in den entsprechenden Abschnitten der Dokumentation, die einem bestimmten Paket gewidmet sind:
Um Skripte sowie alle Modelle zu verwenden, müssen Sie das Repository klonen und Abhängigkeiten installieren:
git clone [email protected]:osmr/imgclsmob.git
pip install -r requirements.txt
Einige Bemerkungen:
Repo ist ein Autor -Repository, wenn es existiert.a , b , c , d und e bedeutet die Implementierung eines Modells für ImageNET-1K, CIFAR-10, CIFAR-100, SVHN und CUB-200-2011.A , B , C , D und E bedeutet, ein vorgebildetes Modell für entsprechende Datensätze zu haben.| Modell | Gluon | Pytorch | Kette | Keras | Tf | Tf2 | Papier | Repo | Jahr |
|---|---|---|---|---|---|---|---|---|---|
| Alexnet | A | A | A | A | A | A | Link | Link | 2012 |
| Zfnet | A | A | A | A | A | A | Link | - - | 2013 |
| VGG | A | A | A | A | A | A | Link | - - | 2014 |
| BN-VGG | A | A | A | A | A | A | Link | - - | 2015 |
| Bn-Inception | A | A | A | - - | - - | A | Link | - - | 2015 |
| Resnet | Abcde | Abcde | Abcde | A | A | Abcde | Link | Link | 2015 |
| Vorrerenet | ABCD | ABCD | ABCD | A | A | ABCD | Link | Link | 2016 |
| Resnext | ABCD | ABCD | ABCD | A | A | ABCD | Link | Link | 2016 |
| Senet | A | A | A | A | A | A | Link | Link | 2017 |
| Se-resnet | Abcde | Abcde | Abcde | A | A | Abcde | Link | Link | 2017 |
| Se-furesnet | ABCD | ABCD | ABCD | A | A | ABCD | Link | Link | 2017 |
| Se-resnext | A | A | A | A | A | A | Link | Link | 2017 |
| Resnest (a) | A | A | A | - - | - - | A | Link | Link | 2020 |
| Ibn-resnet | A | A | - - | - - | - - | A | Link | Link | 2018 |
| Ibn-resnext | A | A | - - | - - | - - | A | Link | Link | 2018 |
| Ibn-Densenet | A | A | - - | - - | - - | A | Link | Link | 2018 |
| Airnet | A | A | A | - - | - - | A | Link | Link | 2018 |
| Airnext | A | A | A | - - | - - | A | Link | Link | 2018 |
| BAM-RESNET | A | A | A | - - | - - | A | Link | Link | 2018 |
| Cbam-resnet | A | A | A | - - | - - | A | Link | Link | 2018 |
| Resattnet | A | A | A | - - | - - | - - | Link | Link | 2017 |
| SKNET | A | A | A | - - | - - | - - | Link | Link | 2019 |
| SCNET | A | A | A | - - | - - | A | Link | Link | 2020 |
| Regnet | A | A | A | - - | - - | A | Link | Link | 2020 |
| DIA-RESNET | ABCD | ABCD | ABCD | - - | - - | - - | Link | Link | 2019 |
| Dia-Presnet | ABCD | ABCD | ABCD | - - | - - | - - | Link | Link | 2019 |
| Pyramidnet | ABCD | ABCD | ABCD | - - | - - | ABCD | Link | Link | 2016 |
| Diracnetv2 | A | A | A | - - | - - | A | Link | Link | 2017 |
| Sharesnet | A | A | A | - - | - - | - - | Link | Link | 2017 |
| Cru-net | A | - - | - - | - - | - - | - - | Link | Link | 2018 |
| Densenet | ABCD | ABCD | ABCD | A | A | ABCD | Link | Link | 2016 |
| Kondensenset | A | A | A | - - | - - | - - | Link | Link | 2017 |
| Sparsenet | A | A | A | - - | - - | - - | Link | Link | 2018 |
| Peleenet | A | A | A | - - | - - | A | Link | Link | 2018 |
| Oktober-RESNET | ABCD | A | A | - - | - - | - - | Link | - - | 2019 |
| Res2net | A | - - | - - | - - | - - | - - | Link | - - | 2019 |
| Wrn | ABCD | ABCD | ABCD | - - | - - | A | Link | Link | 2016 |
| WRN-1bit | BCD | BCD | BCD | - - | - - | - - | Link | Link | 2018 |
| DRN-C | A | A | A | - - | - - | A | Link | Link | 2017 |
| Drn-d | A | A | A | - - | - - | A | Link | Link | 2017 |
| DPN | A | A | A | - - | - - | A | Link | Link | 2017 |
| Darknet Ref | A | A | A | A | A | A | Link | Link | - - |
| Darknet Tiny | A | A | A | A | A | A | Link | Link | - - |
| Darknet-19 | A | A | A | A | A | A | Link | Link | - - |
| Darknet-53 | A | A | A | A | A | A | Link | Link | 2018 |
| ChannelNet | A | A | A | - - | A | - - | Link | Link | 2018 |
| ISQRT-CoV-RESNET | A | A | - - | - - | - - | - - | Link | Link | 2017 |
| Revnet | - - | A | - - | - - | - - | - - | Link | Link | 2017 |
| i-revnet | A | A | A | - - | - - | - - | Link | Link | 2018 |
| Bagnet | A | A | A | - - | - - | A | Link | Link | 2019 |
| DLA | A | A | A | - - | - - | A | Link | Link | 2017 |
| Msdnet | A | ab | - - | - - | - - | - - | Link | Link | 2017 |
| Fischerei | A | A | A | - - | - - | - - | Link | Link | 2018 |
| ESPNETV2 | A | A | A | - - | - - | - - | Link | Link | 2018 |
| Dizenet | A | A | A | - - | - - | A | Link | Link | 2019 |
| Hrnet | A | A | A | - - | - - | A | Link | Link | 2019 |
| Vovnet | A | A | A | - - | - - | A | Link | Link | 2019 |
| Selecsls | A | A | A | - - | - - | A | Link | Link | 2019 |
| Hardnet | A | A | A | - - | - - | A | Link | Link | 2019 |
| X-Densenet | ABCD | ABCD | ABCD | - - | - - | - - | Link | Link | 2017 |
| Squeezenet | A | A | A | A | A | A | Link | Link | 2016 |
| Squeezeresnet | A | A | A | A | A | A | Link | - - | 2016 |
| Squeezenext | A | A | A | A | A | A | Link | Link | 2018 |
| Mischungen | A | A | A | A | A | A | Link | - - | 2017 |
| SHUFFLENETV2 | A | A | A | A | A | A | Link | - - | 2018 |
| Menet | A | A | A | A | A | A | Link | Link | 2018 |
| Mobilenet | Ae | Ae | Ae | A | A | Ae | Link | Link | 2017 |
| FD-Mobilenet | A | A | A | A | A | A | Link | Link | 2018 |
| Mobilenetv2 | A | A | A | A | A | A | Link | Link | 2018 |
| Mobilenetv3 | A | A | A | A | - - | A | Link | Link | 2019 |
| Igcv3 | A | A | A | A | A | A | Link | Link | 2018 |
| Ghostnet | A | A | A | - - | - - | A | Link | Link | 2019 |
| Mnasnet | A | A | A | A | A | A | Link | - - | 2018 |
| DARTS | A | A | A | - - | - - | - - | Link | Link | 2018 |
| Proxylessnas | Ae | Ae | Ae | - - | - - | Ae | Link | Link | 2018 |
| FBNET-C | A | A | A | - - | - - | A | Link | - - | 2018 |
| Xception | A | A | A | - - | - - | A | Link | Link | 2016 |
| Inceptionv3 | A | A | A | - - | - - | A | Link | Link | 2015 |
| Inceptionv4 | A | A | A | - - | - - | A | Link | Link | 2016 |
| InceptionResnetv1 | A | A | A | - - | - - | A | Link | Link | 2016 |
| InceptionResnetv2 | A | A | A | - - | - - | A | Link | Link | 2016 |
| Polynet | A | A | A | - - | - - | A | Link | Link | 2016 |
| Nasnet-Large | A | A | A | - - | - - | A | Link | Link | 2017 |
| Nasnet-Mobile | A | A | A | - - | - - | A | Link | Link | 2017 |
| Pnasnet-large | A | A | A | - - | - - | A | Link | Link | 2017 |
| Spnasnet | A | A | A | - - | - - | A | Link | Link | 2019 |
| EffizientesNetz | A | A | A | A | - - | A | Link | Link | 2019 |
| Mixnet | A | A | A | - - | - - | A | Link | Link | 2019 |
| Nin | BCD | BCD | BCD | - - | - - | - - | Link | Link | 2013 |
| ROR-3 | BCD | BCD | BCD | - - | - - | - - | Link | - - | 2016 |
| Rir | BCD | BCD | BCD | - - | - - | - - | Link | - - | 2016 |
| Resdrop-resnet | BCD | BCD | BCD | - - | - - | - - | Link | Link | 2016 |
| Shake-Shake-Resnet | BCD | BCD | BCD | - - | - - | - - | Link | Link | 2017 |
| Shakedrop-resnet | BCD | BCD | BCD | - - | - - | - - | Link | - - | 2018 |
| Fractalnet | BC | BC | - - | - - | - - | - - | Link | Link | 2016 |
| NTS-NET | E | E | E | - - | - - | - - | Link | Link | 2018 |
Einige Bemerkungen:
a/A entspricht Pascal VOC2012.b/B entspricht ADE20K.c/C entspricht Stadtlandschaften.d/D entspricht Coco.e/E entspricht Celebamask-hq.| Modell | Gluon | Pytorch | Kette | Keras | Tf | Tf2 | Papier | Repo | Jahr |
|---|---|---|---|---|---|---|---|---|---|
| PSPNET | ABCD | ABCD | ABCD | - - | - - | ABCD | Link | - - | 2016 |
| Deeptlabv3 | ABCD | ABCD | ABCD | - - | - - | ABCD | Link | - - | 2017 |
| FCN-8S (D) | ABCD | ABCD | ABCD | - - | - - | ABCD | Link | - - | 2014 |
| ICnet | C | C | C | - - | - - | C | Link | Link | 2017 |
| Sinet | C | C | C | - - | - - | C | Link | Link | 2019 |
| Bisenet | e | e | e | - - | - - | e | Link | - - | 2018 |
| Danet | C | C | C | - - | - - | C | Link | Link | 2018 |
| Schnell-Scnn | C | C | C | - - | - - | C | Link | - - | 2019 |
| CGGET | C | C | C | - - | - - | C | Link | Link | 2018 |
| Dabnet | C | C | C | - - | - - | C | Link | Link | 2019 |
| Fpenet | C | C | C | - - | - - | C | Link | - - | 2019 |
| Kontextnetz | - - | C | - - | - - | - - | - - | Link | - - | 2018 |
| Lednet | C | C | C | - - | - - | C | Link | - - | 2019 |
| Esnet | - - | C | - - | - - | - - | - - | Link | - - | 2019 |
| Edanet | - - | C | - - | - - | - - | - - | Link | Link | 2018 |
| Enet | - - | C | - - | - - | - - | - - | Link | - - | 2016 |
| Erfnet | - - | C | - - | - - | - - | - - | Link | - - | 2017 |
| LinkNet | - - | C | - - | - - | - - | - - | Link | - - | 2017 |
| Segnet | - - | C | - - | - - | - - | - - | Link | - - | 2015 |
| U-net | - - | C | - - | - - | - - | - - | Link | - - | 2015 |
| SQNET | - - | C | - - | - - | - - | - - | Link | - - | 2016 |
Einige Bemerkungen:
a/A entspricht Coco.| Modell | Gluon | Pytorch | Kette | Keras | Tf | Tf2 | Papier | Repo | Jahr |
|---|---|---|---|---|---|---|---|---|---|
| Centernet | A | A | A | - - | - - | A | Link | Link | 2019 |
Einige Bemerkungen:
a/A entspricht Coco.| Modell | Gluon | Pytorch | Kette | Keras | Tf | Tf2 | Papier | Repo | Jahr |
|---|---|---|---|---|---|---|---|---|---|
| Alphapose | A | A | A | - - | - - | A | Link | Link | 2016 |
| Einfacher | A | A | A | - - | - - | A | Link | Link | 2018 |
| Simplepell (Mobile) | A | A | A | - - | - - | A | Link | - - | 2018 |
| Leichtes Gewicht | A | A | A | - - | - - | A | Link | Link | 2018 |
| Ibppose | A | A | A | - - | - - | A | Link | Link | 2019 |
Einige Bemerkungen:
a/A entspricht Librispeech.b/B entspricht Mozilla Common Voice.| Modell | Gluon | Pytorch | Kette | Keras | Tf | Tf2 | Papier | Repo | Jahr |
|---|---|---|---|---|---|---|---|---|---|
| Jasper Dr. | Ab | Ab | ab | - - | - - | ab | Link | Link | 2019 |
| Quarznet | Ab | Ab | ab | - - | - - | ab | Link | Link | 2019 |