imgclsmob
1.0.0
該存儲庫用於研究卷積網絡主要用於計算機視覺任務。為此,存儲庫包含(重新)實現各種分類,分割,檢測和姿勢估計模型和腳本,以培訓/評估/轉換。
使用以下框架:
對於每個受支持的框架,都有一個包含沒有輔助腳本的純模型的PIP包裝。包裝列表:
當前,模型主要是在Gluon上實現的,然後移植到其他框架上。某些型號在Imagenet-1K,Cifar-10/100,SVHN,Cub-200-2011,Pascal VOC2012,ADE20K,CityScapes和CoCo數據集上進行了預估計。所有預驗證的重量在使用過程中自動加載。請參閱專用於特定軟件包的文檔的相應部分中的權重自動加載的示例:
要使用培訓/評估腳本以及所有模型,您需要克隆存儲庫並安裝依賴項:
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-20011的模型實現。A , B , C , D和E表示對相應數據集具有預訓練的模型。| 模型 | gluon | Pytorch | 連鎖商 | 凱拉斯 | TF | TF2 | 紙 | 倉庫 | 年 |
|---|---|---|---|---|---|---|---|---|---|
| Alexnet | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | 關聯 | 2012 |
| ZFNET | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | - | 2013 |
| VGG | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | - | 2014 |
| bn-vgg | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | - | 2015 |
| bn啟動 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | - | 2015 |
| 重新連接 | Abcde | Abcde | Abcde | 一個 | 一個 | Abcde | 關聯 | 關聯 | 2015 |
| preesnet | 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 |
| 重新開頭(a) | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2020 |
| IBN-RESNET | 一個 | 一個 | - | - | - | 一個 | 關聯 | 關聯 | 2018 |
| IBN-RESNEXT | 一個 | 一個 | - | - | - | 一個 | 關聯 | 關聯 | 2018 |
| IBN-DENSENET | 一個 | 一個 | - | - | - | 一個 | 關聯 | 關聯 | 2018 |
| 氣網 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2018 |
| airnext | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2018 |
| BAM-RESNET | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2018 |
| CBAM-RESNET | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2018 |
| Resattnet | 一個 | 一個 | 一個 | - | - | - | 關聯 | 關聯 | 2017 |
| Sknet | 一個 | 一個 | 一個 | - | - | - | 關聯 | 關聯 | 2019 |
| scnet | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2020 |
| regnet | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2020 |
| Dia-Resnet | ABCD | ABCD | ABCD | - | - | - | 關聯 | 關聯 | 2019 |
| Dia-preresnet | ABCD | ABCD | ABCD | - | - | - | 關聯 | 關聯 | 2019 |
| 金字塔 | ABCD | ABCD | ABCD | - | - | ABCD | 關聯 | 關聯 | 2016 |
| diracnetv2 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2017 |
| Sharesnet | 一個 | 一個 | 一個 | - | - | - | 關聯 | 關聯 | 2017 |
| CRU-NET | 一個 | - | - | - | - | - | 關聯 | 關聯 | 2018 |
| Densenet | ABCD | ABCD | ABCD | 一個 | 一個 | ABCD | 關聯 | 關聯 | 2016 |
| 冷凝網 | 一個 | 一個 | 一個 | - | - | - | 關聯 | 關聯 | 2017 |
| Sparsenet | 一個 | 一個 | 一個 | - | - | - | 關聯 | 關聯 | 2018 |
| Peleenet | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2018 |
| OCT-RESNET | ABCD | 一個 | 一個 | - | - | - | 關聯 | - | 2019 |
| Res2net | 一個 | - | - | - | - | - | 關聯 | - | 2019 |
| WRN | ABCD | ABCD | ABCD | - | - | 一個 | 關聯 | 關聯 | 2016 |
| WRN-1位 | BCD | BCD | BCD | - | - | - | 關聯 | 關聯 | 2018 |
| DRN-C | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2017 |
| DRN-D | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2017 |
| DPN | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2017 |
| Darknet Ref | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | 關聯 | - |
| Darknet Tiny | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | 關聯 | - |
| Darknet-19 | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | 關聯 | - |
| Darknet-53 | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | 關聯 | 2018 |
| Channelnet | 一個 | 一個 | 一個 | - | 一個 | - | 關聯 | 關聯 | 2018 |
| ISQRT-COV-RESNET | 一個 | 一個 | - | - | - | - | 關聯 | 關聯 | 2017 |
| Revnet | - | 一個 | - | - | - | - | 關聯 | 關聯 | 2017 |
| i-revnet | 一個 | 一個 | 一個 | - | - | - | 關聯 | 關聯 | 2018 |
| 袋裝 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2019 |
| DLA | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2017 |
| msdnet | 一個 | ab | - | - | - | - | 關聯 | 關聯 | 2017 |
| 魚網 | 一個 | 一個 | 一個 | - | - | - | 關聯 | 關聯 | 2018 |
| ESPNETV2 | 一個 | 一個 | 一個 | - | - | - | 關聯 | 關聯 | 2018 |
| dicenet | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2019 |
| hrnet | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2019 |
| vovnet | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2019 |
| Selecsls | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2019 |
| 硬網 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2019 |
| X-Densenet | ABCD | ABCD | ABCD | - | - | - | 關聯 | 關聯 | 2017 |
| 擠壓 | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | 關聯 | 2016 |
| 擠壓器 | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | - | 2016 |
| Squeezenext | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | 關聯 | 2018 |
| Shufflenet | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | - | 2017 |
| ShuffLenetV2 | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | - | 2018 |
| Menet | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | 關聯 | 2018 |
| Mobilenet | Ae | Ae | Ae | 一個 | 一個 | Ae | 關聯 | 關聯 | 2017 |
| FD-Mobilenet | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | 關聯 | 2018 |
| Mobilenetv2 | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | 關聯 | 2018 |
| Mobilenetv3 | 一個 | 一個 | 一個 | 一個 | - | 一個 | 關聯 | 關聯 | 2019 |
| IGCV3 | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | 關聯 | 2018 |
| Ghostnet | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2019 |
| mnasnet | 一個 | 一個 | 一個 | 一個 | 一個 | 一個 | 關聯 | - | 2018 |
| 飛鏢 | 一個 | 一個 | 一個 | - | - | - | 關聯 | 關聯 | 2018 |
| 近親納 | Ae | Ae | Ae | - | - | Ae | 關聯 | 關聯 | 2018 |
| fbnet-c | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | - | 2018 |
| X Ception | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2016 |
| Inceptionv3 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2015 |
| Inceptionv4 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2016 |
| InceptionResnetv1 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2016 |
| InceptionResnetv2 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2016 |
| 多網 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2016 |
| Nasnet-large | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2017 |
| NASNET-MOBILE | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2017 |
| pnasnet-large | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2017 |
| spnasnet | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2019 |
| 有效網絡 | 一個 | 一個 | 一個 | 一個 | - | 一個 | 關聯 | 關聯 | 2019 |
| 混合網 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2019 |
| 寧 | BCD | BCD | BCD | - | - | - | 關聯 | 關聯 | 2013 |
| ROR-3 | BCD | BCD | BCD | - | - | - | 關聯 | - | 2016 |
| RIR | BCD | BCD | BCD | - | - | - | 關聯 | - | 2016 |
| recdrop-resnet | BCD | BCD | BCD | - | - | - | 關聯 | 關聯 | 2016 |
| 搖動振動架 | BCD | BCD | BCD | - | - | - | 關聯 | 關聯 | 2017 |
| shakedrop-resnet | BCD | BCD | BCD | - | - | - | 關聯 | - | 2018 |
| 分形 | 公元前 | 公元前 | - | - | - | - | 關聯 | 關聯 | 2016 |
| NTS-NET | e | e | e | - | - | - | 關聯 | 關聯 | 2018 |
一些評論:
a/A對應於Pascal VOC2012。b/B對應於ADE20K。c/C對應於城市景觀。d/D對應於可可。e/E對應於Celebamask-HQ。| 模型 | gluon | Pytorch | 連鎖商 | 凱拉斯 | TF | TF2 | 紙 | 倉庫 | 年 |
|---|---|---|---|---|---|---|---|---|---|
| pspnet | ABCD | ABCD | ABCD | - | - | ABCD | 關聯 | - | 2016 |
| DeepLabv3 | ABCD | ABCD | ABCD | - | - | ABCD | 關聯 | - | 2017 |
| FCN-8S(D) | ABCD | ABCD | ABCD | - | - | ABCD | 關聯 | - | 2014 |
| ICNET | c | c | c | - | - | c | 關聯 | 關聯 | 2017 |
| 辛納特 | c | c | c | - | - | c | 關聯 | 關聯 | 2019 |
| Bisenet | e | e | e | - | - | e | 關聯 | - | 2018 |
| 丹特 | c | c | c | - | - | c | 關聯 | 關聯 | 2018 |
| 快速掃描 | c | c | c | - | - | c | 關聯 | - | 2019 |
| CGNET | c | c | c | - | - | c | 關聯 | 關聯 | 2018 |
| dabnet | c | c | c | - | - | c | 關聯 | 關聯 | 2019 |
| fpenet | c | c | c | - | - | c | 關聯 | - | 2019 |
| 上下文網 | - | c | - | - | - | - | 關聯 | - | 2018 |
| Lednet | c | c | c | - | - | c | 關聯 | - | 2019 |
| esnet | - | c | - | - | - | - | 關聯 | - | 2019 |
| Edanet | - | c | - | - | - | - | 關聯 | 關聯 | 2018 |
| enet | - | c | - | - | - | - | 關聯 | - | 2016 |
| erfnet | - | c | - | - | - | - | 關聯 | - | 2017 |
| linknet | - | c | - | - | - | - | 關聯 | - | 2017 |
| segnet | - | c | - | - | - | - | 關聯 | - | 2015 |
| U-net | - | c | - | - | - | - | 關聯 | - | 2015 |
| SQNET | - | c | - | - | - | - | 關聯 | - | 2016 |
一些評論:
a/A對應於可可。| 模型 | gluon | Pytorch | 連鎖商 | 凱拉斯 | TF | TF2 | 紙 | 倉庫 | 年 |
|---|---|---|---|---|---|---|---|---|---|
| 百分點 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2019 |
一些評論:
a/A對應於可可。| 模型 | gluon | Pytorch | 連鎖商 | 凱拉斯 | TF | TF2 | 紙 | 倉庫 | 年 |
|---|---|---|---|---|---|---|---|---|---|
| 字母 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2016 |
| 簡單 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2018 |
| 簡單(移動) | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | - | 2018 |
| 輕巧的開放點 | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2018 |
| ibppose | 一個 | 一個 | 一個 | - | - | 一個 | 關聯 | 關聯 | 2019 |
一些評論:
a/A對應於LibrisPeech。b/B對應於Mozilla的通用聲音。| 模型 | gluon | Pytorch | 連鎖商 | 凱拉斯 | TF | TF2 | 紙 | 倉庫 | 年 |
|---|---|---|---|---|---|---|---|---|---|
| 賈斯珀博士 | ab | ab | ab | - | - | ab | 關聯 | 關聯 | 2019 |
| Quartznet | ab | ab | ab | - | - | ab | 關聯 | 關聯 | 2019 |