深度学习网络
这个仓库主要用于研究计算机视觉任务中的卷积网络。为此,仓库包含了各种分类、分割、检测和姿态估计模型的重新实现和脚本,用于训练/评估/转换。
使用了以下框架:
对于每个支持的框架,有一个 PIP 包仅包含纯模型而无辅助脚本。包的列表:
- gluoncv2 适用于 Gluon,
- pytorchcv 适用于 PyTorch,
- chainercv2 适用于 Chainer,
- kerascv 适用于 Keras,
- tensorflowcv 适用于 TensorFlow 1.x,
- tf2cv 适用于 TensorFlow 2.x.
目前,大部分模型首先在 Gluon 上实现,然后移植到其他框架。某些模型在ImageNet-1K, CIFAR-10/100, SVHN, CUB-200-2011, Pascal VOC2012, ADE20K, Cityscapes, 和 COCO 数据集上预训练。所有预训练权重在使用时会自动加载。在文档中专门针对特定包的部分,可以找到这些权重自动加载的示例:
安装
要使用训练/评估脚本以及所有模型,你需要克隆该仓库并安装依赖项:
git clone git@github.com:osmr/imgclsmob.git
pip install -r requirements.txt
已实现分类模型表
一些备注:
Repo
是作者仓库(如果存在)。a
、b
、c
、d
和e
分别表示 ImageNet-1K、CIFAR-10、CIFAR-100、SVHN 和 CUB-200-2011 模型的实现。A
、B
、C
、D
和E
分别表示相应数据集的预训练模型。 | 模型 | Gluon | PyTorch | Chainer | Keras | TF | TF2 | 论文 | 库 | 年份 | | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | | AlexNet | A | A | A | A | A | A | 链接 | 链接 | 2012 | | ZFNet | A | A | A | A | A | A | 链接 | - | 2013 | | VGG | A | A | A | A | A | A | 链接 | - | 2014 | | BN-VGG | A | A | A | A | A | A | 链接 | - | 2015 | | BN-Inception | A | A | A | - | - | A | 链接 | - | 2015 | | ResNet | ABCDE | ABCDE | ABCDE | A | A | ABCDE | 链接 | 链接 | 2015 | | PreResNet | ABCD | ABCD | ABCD | A | A | ABCD | 链接 | 链接 | 2016 | | ResNeXt | ABCD | ABCD | ABCD | A | A | ABCD | 链接 | 链接 | 2016 | | SENet | A | A | A | A | A | A | 链接 | 链接 | 2017 | | SE-ResNet | ABCDE | ABCDE | ABCDE | A | A | ABCDE | 链接 | 链接 | 2017 | | SE-PreResNet | ABCD | ABCD | ABCD | A | A | ABCD | 链接 | 链接 | 2017 | | SE-ResNeXt | A | A | A | A | A | A | 链接 | 链接 | 2017 | | ResNeSt(A) | A | A | A | - | - | A | 链接 | 链接 | 2020 | | IBN-ResNet | A | A | - | - | - | A | 链接 | 链接 | 2018 | | IBN-ResNeXt | A | A | - | - | - | A | 链接 | 链接 | 2018 | | IBN-DenseNet | A | A | - | - | - | A | 链接 | 链接 | 2018 | | AirNet | A | A | A | - | - | A | 链接 | 链接 | 2018 | | AirNeXt | A | A | A | - | - | A | 链接 | 链接 | 2018 | | BAM-ResNet | A | A | A | - | - | A | 链接 | 链接 | 2018 | | CBAM-ResNet | A | A | A | - | - | A | 链接 | 链接 | 2018 | | ResAttNet | a | a | a | - | - | - | 链接 | 链接 | 2017 | | SKNet | a | a | a | - | - | - | 链接 | 链接 | 2019 | | SCNet | A | A | A | - | - | A | 链接 | 链接 | 2020 | | RegNet | A | A | A | - | - | A | 链接 | 链接 | 2020 | | DIA-ResNet | aBCD | aBCD | aBCD | - | - | - | 链接 | 链接 | 2019 | | DIA-PreResNet | aBCD | aBCD | aBCD | - | - | - | 链接 | 链接 | 2019 | | PyramidNet | ABCD | ABCD | ABCD | - | - | ABCD | 链接 | 链接 | 2016 | | DiracNetV2 | A | A | A | - | - | A | 链接 | 链接 | 2017 | | ShaResNet | a | a | a | - | - | - | 链接 | 链接 | 2017 | | CRU-Net | A | - | - | - | - | - | 链接 | 链接 | 2018 | | DenseNet | ABCD | ABCD | ABCD | A | A | ABCD | 链接 | 链接 | 2016 | | CondenseNet | A | A | A | - | - | - | 链接 | 链接 | 2017 | | SparseNet | a | a | a | - | - | - | 链接 | 链接 | 2018 | | PeleeNet | A | A | A | - | - | A | 链接 | 链接 | 2018 | | Oct-ResNet | abcd | a | a | - | - | - | 链接 | - | 2019 | | Res2Net | a | - | - | - | - | - | 链接 | - | 2019 | | WRN | ABCD | ABCD | ABCD | - | - | a | 链接 | 链接 | 2016 | | WRN-1bit | BCD | BCD | BCD | - | - | - | 链接 | 链接 | 2018 | | DRN-C | A | A | A | - | - | A | 链接 | 链接 | 2017 | | DRN-D | A | A | A | - | - | A | 链接 | 链接 | 2017 | | DPN | A | A | A | - | - | A | 链接 | 链接 | 2017 | | DarkNet Ref | A | A | A | A | A | A | 链接 | 链接 | - | | DarkNet Tiny | A | A | A | A | A | A | 链接 | 链接 | - | | DarkNet-19 | a | a | a | a | a | a | 链接 | 链接 | - | | DarkNet-53 | A | A | A | A | A | A | 链接 | 链接 | 2018 | | ChannelNet | a | a | a | - | a | - | 链接 | 链接 | 2018 | | iSQRT-COV-ResNet | a | a | - | - | - | - | 链接 | 链接 | 2017 | | RevNet | - | a | - | - | - | - | 链接 | 链接 | 2017 | | i-RevNet | A | A | A | - | - | - | 链接 | 链接 | 2018 | | BagNet | A | A | A | - | - | A | 链接 | 链接 | 2019 | | DLA | A | A | A | - | - | A | 链接 | 链接 | 2017 | | MSDNet | a | ab | - | - | - | - | 链接 | 链接 | 2017 | | FishNet | A | A | A | - | - | - | 链接 | 链接 | 2018 | | ESPNetv2 | A | A | A | - | - | - | 链接 | 链接 | 2018 | | DiCENet | A | A | A | - | - | A | 链接 | 链接 | 2019 | | HRNet | A | A | A | - | - | A | 链接 | 链接 | 2019 | | VoVNet | A | A | A | - | - | A | 链接 | 链接 | 2019 | | SelecSLS | A | A | A | - | - | A | 链接 | 链接 | 2019 | | HarDNet | A | A | A | - | - | A | 链接 | 链接 | 2019 | | X-DenseNet | aBCD | aBCD | aBCD | - | - | - | 链接 | 链接 | 2017 | | SqueezeNet | A | A | A | A | A | A | 链接 | 链接 | 2016 | | SqueezeResNet | A | A | A | A | A | A | 链接 | - | 2016 | | SqueezeNext | A | A | A | A | A | A | 链接 | 链接 | 2018 | | ShuffleNet | A | A | A | A | A | A | 链接 | - | 2017 | | ShuffleNetV2 | A | A | A | A | A | A | 链接 | - | 2018 | | MENet | A | A | A | A | A | A | 链接 | 链接 | 2018 | | MobileNet | AE | AE | AE | A | A | AE | 链接 | 链接 | 2017 | | FD-MobileNet | A | A | A | A | A | A | 链接 | 链接 | 2018 | | MobileNetV2 | A | A | A | A | A | A | 链接 | 链接 | 2018 | | MobileNetV3 | A | A | A | A | - | A | 链接 | 链接 | 2019 | | IGCV3 | A | A | A | A | A | A | 链接 | 链接 | 2018 | | GhostNet | a | a | a | - | - | a | 链接 | 链接 | 2019 | | MnasNet | A | A | A | A | A | A | 链接 | - | 2018 | | DARTS | A | A | A | - | - | - | 链接 | 链接 | 2018 | | ProxylessNAS | AE | AE | AE | - | - | AE | 链接 | 链接 | 2018 | | FBNet-C | A | A | A | - | - | A | 链接 | - | 2018 | | Xception | A | A | A | - | - | A | 链接 | 链接 | 2016 | | InceptionV3 | A | A | A | - | - | A | 链接 | 链接 | 2015 | | InceptionV4 | A | A | A | - | - | A | 链接 | 链接 | 2016 | | InceptionResNetV1 | A | A | A | - | - | A | 链接 | 链接 | 2016 | | InceptionResNetV2 | A | A | A | - | - | A | 链接 | 链接 | 2016 | | PolyNet | A | A | A | - | - | A | 链接 | 链接 | 2016 | | NASNet-Large | A | A | A | - | - | A | 链接 | 链接 | 2017 | | NASNet-Mobile | A | A | A | - | - | A | 链接 | 链接 | 2017 | | PNASNet-Large | A | A | A | - | - | A | 链接 | 链接 | 2017 | | SPNASNet | A | A | A | - | - | A | 链接 | 链接 | 2019 | | EfficientNet | A | A | A | A | - | A | 链接 | 链接 | 2019 | | MixNet | A | A | A | - | - | A | 链接 | 链接 | 2019 | | NIN | BCD | BCD | BCD | - | - | - | 链接 | 链接 | 2013 | | RoR-3 | BCD | BCD | BCD | - | - | - | 链接 | - | 2016 | | RiR | BCD | BCD | BCD | - | - | - | 链接 | - | 2016 | | ResDrop-ResNet | bcd | bcd | bcd | - | - | - | 链接 | 链接 | 2016 | | Shake-Shake-ResNet | BCD | BCD | BCD | - | - | - | 链接 | 链接 | 2017 | | ShakeDrop-ResNet | bcd | bcd | bcd | - | - | - | 链接 | - | 2018 | | FractalNet | bc | bc | - | - | - | - | 链接 | 链接 | 2016 | | NTS-Net | E | E | E | - | - | - | 链接 | 链接 | 2018 |
已实现的分割模型表
一些说明:
a/A
对应Pascal VOC2012。b/B
对应ADE20K。c/C
对应Cityscapes。d/D
对应COCO。e/E
对应CelebAMask-HQ。
模型 | Gluon | PyTorch | Chainer | Keras | 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 |
SINet | C | C | C | - | - | c | 链接 | 链接 | 2019 |
BiSeNet | e | e | e | - | - | e | 链接 | - | 2018 |
DANet | C | C | C | - | - | C | 链接 | 链接 | 2018 |
Fast-SCNN | C | C | C | - | - | C | 链接 | - | 2019 |
CGNet | c | c | c | - | - | c | 链接 | 链接 | 2018 |
DABNet | c | c | c | - | - | c | 链接 | 链接 | 2019 |
FPENet | c | c | c | - | - | c | 链接 | - | 2019 |
ContextNet | - | 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
对应COCO。
已实现的人体姿态估计模型表
一些说明:
a/A
对应COCO。
模型 | Gluon | PyTorch | Chainer | Keras | TF | TF2 | 论文 | 仓库 | 年份 |
---|---|---|---|---|---|---|---|---|---|
AlphaPose | A | A | A | - | - | A | 链接 | 链接 | 2016 |
SimplePose | A | A | A | - | - | A | 链接 | 链接 | 2018 |
SimplePose(Mobile) | A | A | A | - | - | A | 链接 | - | 2018 |
Lightweight OpenPose | A | A | A | - | - | A | 链接 | 链接 | 2018 |
IBPPose | A | A | A | - | - | A | 链接 | 链接 | 2019 |
已实现的自动语音识别模型表
一些说明:
a/A
对应LibriSpeech。b/B
对应Mozilla Common Voice。