mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
a39cc43374
* Split MobileNetV3 and EfficientNet model files and put builder and blocks in own files (getting too large) * Finalize CondConv EfficientNet variant * Add the AdvProp weights files and B8 EfficientNet model * Refine the feature extraction module for EfficientNet and MobileNetV3
23 lines
608 B
Python
23 lines
608 B
Python
from .inception_v4 import *
|
|
from .inception_resnet_v2 import *
|
|
from .densenet import *
|
|
from .resnet import *
|
|
from .dpn import *
|
|
from .senet import *
|
|
from .xception import *
|
|
from .nasnet import *
|
|
from .pnasnet import *
|
|
from .gen_efficientnet import *
|
|
from .mobilenetv3 import *
|
|
from .inception_v3 import *
|
|
from .gluon_resnet import *
|
|
from .gluon_xception import *
|
|
from .res2net import *
|
|
from .dla import *
|
|
from .hrnet import *
|
|
|
|
from .registry import *
|
|
from .factory import create_model
|
|
from .helpers import load_checkpoint, resume_checkpoint
|
|
from .test_time_pool import TestTimePoolHead, apply_test_time_pool
|