lostkevin 31897984d8
implement onnx export for inception3/4, resnext, mobilenetv2 (#346)
* add inceptionv4 backbone/training settings
* add converted backbone, top-1 acc 80.08
2024-07-18 16:52:56 +08:00

32 lines
1.2 KiB
Python

# Copyright (c) Alibaba, Inc. and its affiliates.
from .benchmark_mlp import BenchMarkMLP
from .bninception import BNInception
from .clip_bert import ClipBertClassification
from .conv_mae_vit import FastConvMAEViT
from .conv_vitdet import ConvViTDet
from .edgevit import EdgeVit
from .efficientformer import EfficientFormer
from .face_keypoint_backbone import FaceKeypointBackbone
from .genet import PlainNet
from .hrnet import HRNet
from .inceptionv3 import Inception3
from .inceptionv4 import Inception4
from .lighthrnet import LiteHRNet
from .mae_vit_transformer import *
from .mit import MixVisionTransformer
from .mnasnet import MNASNet
from .mobilenetv2 import MobileNetV2
from .pytorch_image_models_wrapper import *
from .repvgg_yolox_backbone import RepVGGYOLOX
from .resnest import ResNeSt
from .resnet import ResNet
from .resnet_jit import ResNetJIT
from .resnext import ResNeXt
from .shuffle_transformer import ShuffleTransformer
from .stdc import STDCContextPathNet, STDCNet
from .swin_transformer import SwinTransformer
from .swin_transformer3d import SwinTransformer3D
from .vision_transformer import VisionTransformer
from .vitdet import ViTDet
from .x3d import X3D