Repo ini digunakan untuk meneliti jaringan konvolusional terutama untuk tugas visi komputer. Untuk tujuan ini, repo berisi (kembali) implementasi dari berbagai klasifikasi, segmentasi, deteksi, dan pose model estimasi dan skrip untuk pelatihan/evaluasi/konversi.
Kerangka kerja berikut digunakan:
Untuk setiap kerangka kerja yang didukung, ada paket pip yang berisi model murni tanpa skrip tambahan. Daftar Paket:
Saat ini, model sebagian besar diimplementasikan di Gluon dan kemudian diangkut ke kerangka kerja lainnya. Beberapa model pretrain di ImageNet-1K, CIFAR-10/100, SVHN, CUB-200-2011, Pascal VOC2012, ADE20K, Cityscapes, dan Coco Datasets. Semua bobot pretrained dimuat secara otomatis selama penggunaan. Lihat contoh pemuatan bobot otomatis seperti itu di bagian yang sesuai dari dokumentasi yang didedikasikan untuk paket tertentu:
Untuk menggunakan skrip pelatihan/evaluasi serta semua model, Anda perlu mengkloning repositori dan menginstal dependensi:
git clone [email protected]:osmr/imgclsmob.git
pip install -r requirements.txt
Beberapa komentar:
Repo adalah repositori penulis, jika ada.a , b , c , d , dan e berarti implementasi model untuk ImageNet-1K, CIFAR-10, CIFAR-100, SVHN, dan CUB-200-2011, masing-masing.A , B , C , D , dan E berarti memiliki model pra-terlatih untuk set data yang sesuai.| Model | Gluon | Pytorch | Chainer | Keras | Tf | Tf2 | Kertas | Repo | Tahun |
|---|---|---|---|---|---|---|---|---|---|
| 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 |
| Preresnet | 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-PRERESNET | 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 |
| Restnet | 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-preresnet | ABCD | ABCD | ABCD | - | - | - | link | link | 2019 |
| Piramidnet | 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 |
| Condensenet | A | A | A | - | - | - | link | link | 2017 |
| SPARSENET | A | A | A | - | - | - | link | link | 2018 |
| Pelenet | A | A | A | - | - | A | link | link | 2018 |
| OCT-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 |
| Langgai | A | A | A | - | - | - | link | link | 2018 |
| Espnetv2 | A | A | A | - | - | - | link | link | 2018 |
| Dicenet | 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 |
| Shufflenet | 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 |
| PANAHAN | 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 |
| EfisienNet | 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 |
Beberapa komentar:
a/A sesuai dengan Pascal VOC2012.b/B sesuai dengan ADE20K.c/C sesuai dengan Cityscapes.d/D sesuai dengan Coco.e/E sesuai dengan Celebamask-HQ.| Model | Gluon | Pytorch | Chainer | Keras | Tf | Tf2 | Kertas | Repo | Tahun |
|---|---|---|---|---|---|---|---|---|---|
| Pspnet | ABCD | ABCD | ABCD | - | - | ABCD | link | - | 2016 |
| Deeplabv3 | 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 |
| Scnn cepat | C | C | C | - | - | C | link | - | 2019 |
| Cgnet | C | C | C | - | - | C | link | link | 2018 |
| Dabnet | C | C | C | - | - | C | link | link | 2019 |
| Fpenet | C | C | C | - | - | C | link | - | 2019 |
| ContextNet | - | 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 |
Beberapa komentar:
a/A sesuai dengan Coco.| Model | Gluon | Pytorch | Chainer | Keras | Tf | Tf2 | Kertas | Repo | Tahun |
|---|---|---|---|---|---|---|---|---|---|
| Centernet | A | A | A | - | - | A | link | link | 2019 |
Beberapa komentar:
a/A sesuai dengan Coco.| Model | Gluon | Pytorch | Chainer | Keras | Tf | Tf2 | Kertas | Repo | Tahun |
|---|---|---|---|---|---|---|---|---|---|
| Alfaposa | A | A | A | - | - | A | link | link | 2016 |
| Simplepose | A | A | A | - | - | A | link | link | 2018 |
| SimplePose (Mobile) | A | A | A | - | - | A | link | - | 2018 |
| Openpose ringan | A | A | A | - | - | A | link | link | 2018 |
| Ibppose | A | A | A | - | - | A | link | link | 2019 |
Beberapa komentar:
a/A sesuai dengan Librispeech.b/B sesuai dengan suara umum Mozilla.| Model | Gluon | Pytorch | Chainer | Keras | Tf | Tf2 | Kertas | Repo | Tahun |
|---|---|---|---|---|---|---|---|---|---|
| Jasper DR | AB | AB | AB | - | - | AB | link | link | 2019 |
| Kuarsa | AB | AB | AB | - | - | AB | link | link | 2019 |