Skip to content

computer vision, resnet, resnetv2, densenet, resnext, resnest, efficientnet, dla,mobilenet,shufflenet, shufflenetv2(tensorflow2/tf2)

License

Notifications You must be signed in to change notification settings

wavce/classificationx

Repository files navigation

ClassificationX

Introduction

ClassificaitonX is an open source 2D image classfication toolbox based on Tensorflow2.4+.

Changelog

  • 2021.05.19 add resnest, shufflentv1, shufflenetv2.
  • 2021.01.05 Initialized the repository.

Model

ResNet

model #Params BFLOPs top-1 error top-5 error
ResNet18 30.24 10.92
ResNet34 26.70 8.58
ResNet50 22.73M 9.74 23.85 7.13
ResNet101 22.63 6.44
ResNet152 27.21M 22.6 21.69 5.94

Res2Net

model #Params BFLOPs top-1 error top-5 error link
Res2Net-50-48w-2s 25.29M 4.2 22.68 6.47
Res2Net-50-26w-4s 25.70M 4.2 22.01 6.15
Res2Net-50-14w-8s 25.06M 4.2 21.86 6.14
Res2Net-50-26w-6s 37.05M 6.3 21.42 5.87
Res2Net-50-26w-8s 48.40M 8.3 20.80 5.63
Res2Net-101-26w-4s 45.21M 8.1 20.81 5.57
Res2Net-v1b-50 25.72M 4.5 19.73 4.96
Res2Net-v1b-101 45.23M 8.3 18.77 4.64
Res2NeXt-50 24.67M 4.2 21.76 6.09
Res2Net-DLA-60 21.15M 4.2 21.53 5.80
Res2NeXt-DLA-60 17.33M 3.6 21.55 5.86

CrossStagePartialNetwork

model #Params BFLOPs top-1 error top-5 error link
Darknet-53 41.57M 18.57 22.8 6.2
CSPDarkNet-53 27.61M 13.07 22.8 6.4

ResNeSt

model #Params BFLOPs top-1 error top-5 error link
ResNeSt-50 - - 18.95 5.58 baidu[5fzh]
ResNeSt-101 - - 17.18 3.69 baidu[utbi]
ResNeSt-200 - - 16.16 3.14 baidu[x2jc]
ResNeSt-269 - - 15.57 3.02 baidu[7bap]

ReXNet

model #Params BFLOPs top-1 error top-5 error link
ReXNet_0.9 4.1M 0.35 22.8 6.5
ReXNet_1.0 4.8M 0.40 22.1 6.1 baidu[7j5d]
ReXNet_1.3 7.6M 0.66 20.5 5.3 baidu[knir]
ReXNet_1.5 9.7M 0.88 19.7 4.8 baidu[vev5]
ReXNet_2.0 19M 1.8 18.4 4.3 baidu[ayv6]

About

computer vision, resnet, resnetv2, densenet, resnext, resnest, efficientnet, dla,mobilenet,shufflenet, shufflenetv2(tensorflow2/tf2)

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages