├── .gitignore ├── LICENSE ├── README.md ├── assets ├── dogcat.jpg ├── grids.png ├── head.png ├── heads.png └── rabbit.jpg ├── demo.ipynb ├── imagenet_cls.json ├── setup.py ├── timm ├── __init__.py ├── data │ ├── __init__.py │ ├── auto_augment.py │ ├── config.py │ ├── constants.py │ ├── dataset.py │ ├── dataset_factory.py │ ├── distributed_sampler.py │ ├── loader.py │ ├── mixup.py │ ├── parsers │ │ ├── __init__.py │ │ ├── class_map.py │ │ ├── constants.py │ │ ├── parser.py │ │ ├── parser_factory.py │ │ ├── parser_image_folder.py │ │ ├── parser_image_in_tar.py │ │ ├── parser_image_tar.py │ │ └── parser_tfds.py │ ├── random_erasing.py │ ├── real_labels.py │ ├── tf_preprocessing.py │ ├── transforms.py │ └── transforms_factory.py ├── loss │ ├── __init__.py │ ├── asymmetric_loss.py │ ├── cross_entropy.py │ └── jsd.py ├── models │ ├── __init__.py │ ├── byoanet.py │ ├── byobnet.py │ ├── cait.py │ ├── coat.py │ ├── convit.py │ ├── cspnet.py │ ├── densenet.py │ ├── dla.py │ ├── dpn.py │ ├── efficientnet.py │ ├── efficientnet_blocks.py │ ├── efficientnet_builder.py │ ├── factory.py │ ├── features.py │ ├── ghostnet.py │ ├── gluon_resnet.py │ ├── gluon_xception.py │ ├── hardcorenas.py │ ├── helpers.py │ ├── hrnet.py │ ├── hub.py │ ├── inception_resnet_v2.py │ ├── inception_v3.py │ ├── inception_v4.py │ ├── layers │ │ ├── __init__.py │ │ ├── activations.py │ │ ├── activations_jit.py │ │ ├── activations_me.py │ │ ├── adaptive_avgmax_pool.py │ │ ├── blur_pool.py │ │ ├── bottleneck_attn.py │ │ ├── cbam.py │ │ ├── classifier.py │ │ ├── cond_conv2d.py │ │ ├── config.py │ │ ├── conv2d_same.py │ │ ├── conv_bn_act.py │ │ ├── create_act.py │ │ ├── create_attn.py │ │ ├── create_conv2d.py │ │ ├── create_norm_act.py │ │ ├── drop.py │ │ ├── eca.py │ │ ├── evo_norm.py │ │ ├── gather_excite.py │ │ ├── global_context.py │ │ ├── halo_attn.py │ │ ├── helpers.py │ │ ├── inplace_abn.py │ │ ├── involution.py │ │ ├── lambda_layer.py │ │ ├── linear.py │ │ ├── median_pool.py │ │ ├── mixed_conv2d.py │ │ ├── mlp.py │ │ ├── non_local_attn.py │ │ ├── norm.py │ │ ├── norm_act.py │ │ ├── padding.py │ │ ├── patch_embed.py │ │ ├── pool2d_same.py │ │ ├── selective_kernel.py │ │ ├── separable_conv.py │ │ ├── space_to_depth.py │ │ ├── split_attn.py │ │ ├── split_batchnorm.py │ │ ├── squeeze_excite.py │ │ ├── std_conv.py │ │ ├── swin_attn.py │ │ ├── test_time_pool.py │ │ └── weight_init.py │ ├── levit.py │ ├── mlp_mixer.py │ ├── mobilenetv3.py │ ├── nasnet.py │ ├── nest.py │ ├── nfnet.py │ ├── pit.py │ ├── pnasnet.py │ ├── pruned │ │ ├── ecaresnet101d_pruned.txt │ │ ├── ecaresnet50d_pruned.txt │ │ ├── efficientnet_b1_pruned.txt │ │ ├── efficientnet_b2_pruned.txt │ │ └── efficientnet_b3_pruned.txt │ ├── registry.py │ ├── regnet.py │ ├── res2net.py │ ├── resnest.py │ ├── resnet.py │ ├── resnetv2.py │ ├── rexnet.py │ ├── selecsls.py │ ├── senet.py │ ├── sknet.py │ ├── swin_transformer.py │ ├── tnt.py │ ├── tresnet.py │ ├── twins.py │ ├── vgg.py │ ├── visformer.py │ ├── vision_transformer.py │ ├── vision_transformer_hybrid.py │ ├── vovnet.py │ ├── xception.py │ ├── xception_aligned.py │ └── xcit.py ├── optim │ ├── __init__.py │ ├── adabelief.py │ ├── adafactor.py │ ├── adahessian.py │ ├── adamp.py │ ├── adamw.py │ ├── lookahead.py │ ├── nadam.py │ ├── novograd.py │ ├── nvnovograd.py │ ├── optim_factory.py │ ├── radam.py │ ├── rmsprop_tf.py │ └── sgdp.py ├── scheduler │ ├── __init__.py │ ├── cosine_lr.py │ ├── multistep_lr.py │ ├── plateau_lr.py │ ├── scheduler.py │ ├── scheduler_factory.py │ ├── step_lr.py │ └── tanh_lr.py ├── utils │ ├── __init__.py │ ├── agc.py │ ├── checkpoint_saver.py │ ├── clip_grad.py │ ├── cuda.py │ ├── distributed.py │ ├── jit.py │ ├── log.py │ ├── metrics.py │ ├── misc.py │ ├── model.py │ ├── model_ema.py │ ├── random.py │ └── summary.py └── version.py └── visualizer ├── __init__.py └── visualizer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/README.md -------------------------------------------------------------------------------- /assets/dogcat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/assets/dogcat.jpg -------------------------------------------------------------------------------- /assets/grids.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/assets/grids.png -------------------------------------------------------------------------------- /assets/head.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/assets/head.png -------------------------------------------------------------------------------- /assets/heads.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/assets/heads.png -------------------------------------------------------------------------------- /assets/rabbit.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/assets/rabbit.jpg -------------------------------------------------------------------------------- /demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/demo.ipynb -------------------------------------------------------------------------------- /imagenet_cls.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/imagenet_cls.json -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/setup.py -------------------------------------------------------------------------------- /timm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/__init__.py -------------------------------------------------------------------------------- /timm/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/__init__.py -------------------------------------------------------------------------------- /timm/data/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/auto_augment.py -------------------------------------------------------------------------------- /timm/data/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/config.py -------------------------------------------------------------------------------- /timm/data/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/constants.py -------------------------------------------------------------------------------- /timm/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/dataset.py -------------------------------------------------------------------------------- /timm/data/dataset_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/dataset_factory.py -------------------------------------------------------------------------------- /timm/data/distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/distributed_sampler.py -------------------------------------------------------------------------------- /timm/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/loader.py -------------------------------------------------------------------------------- /timm/data/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/mixup.py -------------------------------------------------------------------------------- /timm/data/parsers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/parsers/__init__.py -------------------------------------------------------------------------------- /timm/data/parsers/class_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/parsers/class_map.py -------------------------------------------------------------------------------- /timm/data/parsers/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/parsers/constants.py -------------------------------------------------------------------------------- /timm/data/parsers/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/parsers/parser.py -------------------------------------------------------------------------------- /timm/data/parsers/parser_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/parsers/parser_factory.py -------------------------------------------------------------------------------- /timm/data/parsers/parser_image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/parsers/parser_image_folder.py -------------------------------------------------------------------------------- /timm/data/parsers/parser_image_in_tar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/parsers/parser_image_in_tar.py -------------------------------------------------------------------------------- /timm/data/parsers/parser_image_tar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/parsers/parser_image_tar.py -------------------------------------------------------------------------------- /timm/data/parsers/parser_tfds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/parsers/parser_tfds.py -------------------------------------------------------------------------------- /timm/data/random_erasing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/random_erasing.py -------------------------------------------------------------------------------- /timm/data/real_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/real_labels.py -------------------------------------------------------------------------------- /timm/data/tf_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/tf_preprocessing.py -------------------------------------------------------------------------------- /timm/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/transforms.py -------------------------------------------------------------------------------- /timm/data/transforms_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/data/transforms_factory.py -------------------------------------------------------------------------------- /timm/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/loss/__init__.py -------------------------------------------------------------------------------- /timm/loss/asymmetric_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/loss/asymmetric_loss.py -------------------------------------------------------------------------------- /timm/loss/cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/loss/cross_entropy.py -------------------------------------------------------------------------------- /timm/loss/jsd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/loss/jsd.py -------------------------------------------------------------------------------- /timm/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/__init__.py -------------------------------------------------------------------------------- /timm/models/byoanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/byoanet.py -------------------------------------------------------------------------------- /timm/models/byobnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/byobnet.py -------------------------------------------------------------------------------- /timm/models/cait.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/cait.py -------------------------------------------------------------------------------- /timm/models/coat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/coat.py -------------------------------------------------------------------------------- /timm/models/convit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/convit.py -------------------------------------------------------------------------------- /timm/models/cspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/cspnet.py -------------------------------------------------------------------------------- /timm/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/densenet.py -------------------------------------------------------------------------------- /timm/models/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/dla.py -------------------------------------------------------------------------------- /timm/models/dpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/dpn.py -------------------------------------------------------------------------------- /timm/models/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/efficientnet.py -------------------------------------------------------------------------------- /timm/models/efficientnet_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/efficientnet_blocks.py -------------------------------------------------------------------------------- /timm/models/efficientnet_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/efficientnet_builder.py -------------------------------------------------------------------------------- /timm/models/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/factory.py -------------------------------------------------------------------------------- /timm/models/features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/features.py -------------------------------------------------------------------------------- /timm/models/ghostnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/ghostnet.py -------------------------------------------------------------------------------- /timm/models/gluon_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/gluon_resnet.py -------------------------------------------------------------------------------- /timm/models/gluon_xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/gluon_xception.py -------------------------------------------------------------------------------- /timm/models/hardcorenas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/hardcorenas.py -------------------------------------------------------------------------------- /timm/models/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/helpers.py -------------------------------------------------------------------------------- /timm/models/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/hrnet.py -------------------------------------------------------------------------------- /timm/models/hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/hub.py -------------------------------------------------------------------------------- /timm/models/inception_resnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/inception_resnet_v2.py -------------------------------------------------------------------------------- /timm/models/inception_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/inception_v3.py -------------------------------------------------------------------------------- /timm/models/inception_v4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/inception_v4.py -------------------------------------------------------------------------------- /timm/models/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/__init__.py -------------------------------------------------------------------------------- /timm/models/layers/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/activations.py -------------------------------------------------------------------------------- /timm/models/layers/activations_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/activations_jit.py -------------------------------------------------------------------------------- /timm/models/layers/activations_me.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/activations_me.py -------------------------------------------------------------------------------- /timm/models/layers/adaptive_avgmax_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/adaptive_avgmax_pool.py -------------------------------------------------------------------------------- /timm/models/layers/blur_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/blur_pool.py -------------------------------------------------------------------------------- /timm/models/layers/bottleneck_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/bottleneck_attn.py -------------------------------------------------------------------------------- /timm/models/layers/cbam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/cbam.py -------------------------------------------------------------------------------- /timm/models/layers/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/classifier.py -------------------------------------------------------------------------------- /timm/models/layers/cond_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/cond_conv2d.py -------------------------------------------------------------------------------- /timm/models/layers/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/config.py -------------------------------------------------------------------------------- /timm/models/layers/conv2d_same.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/conv2d_same.py -------------------------------------------------------------------------------- /timm/models/layers/conv_bn_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/conv_bn_act.py -------------------------------------------------------------------------------- /timm/models/layers/create_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/create_act.py -------------------------------------------------------------------------------- /timm/models/layers/create_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/create_attn.py -------------------------------------------------------------------------------- /timm/models/layers/create_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/create_conv2d.py -------------------------------------------------------------------------------- /timm/models/layers/create_norm_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/create_norm_act.py -------------------------------------------------------------------------------- /timm/models/layers/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/drop.py -------------------------------------------------------------------------------- /timm/models/layers/eca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/eca.py -------------------------------------------------------------------------------- /timm/models/layers/evo_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/evo_norm.py -------------------------------------------------------------------------------- /timm/models/layers/gather_excite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/gather_excite.py -------------------------------------------------------------------------------- /timm/models/layers/global_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/global_context.py -------------------------------------------------------------------------------- /timm/models/layers/halo_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/halo_attn.py -------------------------------------------------------------------------------- /timm/models/layers/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/helpers.py -------------------------------------------------------------------------------- /timm/models/layers/inplace_abn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/inplace_abn.py -------------------------------------------------------------------------------- /timm/models/layers/involution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/involution.py -------------------------------------------------------------------------------- /timm/models/layers/lambda_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/lambda_layer.py -------------------------------------------------------------------------------- /timm/models/layers/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/linear.py -------------------------------------------------------------------------------- /timm/models/layers/median_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/median_pool.py -------------------------------------------------------------------------------- /timm/models/layers/mixed_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/mixed_conv2d.py -------------------------------------------------------------------------------- /timm/models/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/mlp.py -------------------------------------------------------------------------------- /timm/models/layers/non_local_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/non_local_attn.py -------------------------------------------------------------------------------- /timm/models/layers/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/norm.py -------------------------------------------------------------------------------- /timm/models/layers/norm_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/norm_act.py -------------------------------------------------------------------------------- /timm/models/layers/padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/padding.py -------------------------------------------------------------------------------- /timm/models/layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/patch_embed.py -------------------------------------------------------------------------------- /timm/models/layers/pool2d_same.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/pool2d_same.py -------------------------------------------------------------------------------- /timm/models/layers/selective_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/selective_kernel.py -------------------------------------------------------------------------------- /timm/models/layers/separable_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/separable_conv.py -------------------------------------------------------------------------------- /timm/models/layers/space_to_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/space_to_depth.py -------------------------------------------------------------------------------- /timm/models/layers/split_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/split_attn.py -------------------------------------------------------------------------------- /timm/models/layers/split_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/split_batchnorm.py -------------------------------------------------------------------------------- /timm/models/layers/squeeze_excite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/squeeze_excite.py -------------------------------------------------------------------------------- /timm/models/layers/std_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/std_conv.py -------------------------------------------------------------------------------- /timm/models/layers/swin_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/swin_attn.py -------------------------------------------------------------------------------- /timm/models/layers/test_time_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/test_time_pool.py -------------------------------------------------------------------------------- /timm/models/layers/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/layers/weight_init.py -------------------------------------------------------------------------------- /timm/models/levit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/levit.py -------------------------------------------------------------------------------- /timm/models/mlp_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/mlp_mixer.py -------------------------------------------------------------------------------- /timm/models/mobilenetv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/mobilenetv3.py -------------------------------------------------------------------------------- /timm/models/nasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/nasnet.py -------------------------------------------------------------------------------- /timm/models/nest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/nest.py -------------------------------------------------------------------------------- /timm/models/nfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/nfnet.py -------------------------------------------------------------------------------- /timm/models/pit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/pit.py -------------------------------------------------------------------------------- /timm/models/pnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/pnasnet.py -------------------------------------------------------------------------------- /timm/models/pruned/ecaresnet101d_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/pruned/ecaresnet101d_pruned.txt -------------------------------------------------------------------------------- /timm/models/pruned/ecaresnet50d_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/pruned/ecaresnet50d_pruned.txt -------------------------------------------------------------------------------- /timm/models/pruned/efficientnet_b1_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/pruned/efficientnet_b1_pruned.txt -------------------------------------------------------------------------------- /timm/models/pruned/efficientnet_b2_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/pruned/efficientnet_b2_pruned.txt -------------------------------------------------------------------------------- /timm/models/pruned/efficientnet_b3_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/pruned/efficientnet_b3_pruned.txt -------------------------------------------------------------------------------- /timm/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/registry.py -------------------------------------------------------------------------------- /timm/models/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/regnet.py -------------------------------------------------------------------------------- /timm/models/res2net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/res2net.py -------------------------------------------------------------------------------- /timm/models/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/resnest.py -------------------------------------------------------------------------------- /timm/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/resnet.py -------------------------------------------------------------------------------- /timm/models/resnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/resnetv2.py -------------------------------------------------------------------------------- /timm/models/rexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/rexnet.py -------------------------------------------------------------------------------- /timm/models/selecsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/selecsls.py -------------------------------------------------------------------------------- /timm/models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/senet.py -------------------------------------------------------------------------------- /timm/models/sknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/sknet.py -------------------------------------------------------------------------------- /timm/models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/swin_transformer.py -------------------------------------------------------------------------------- /timm/models/tnt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/tnt.py -------------------------------------------------------------------------------- /timm/models/tresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/tresnet.py -------------------------------------------------------------------------------- /timm/models/twins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/twins.py -------------------------------------------------------------------------------- /timm/models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/vgg.py -------------------------------------------------------------------------------- /timm/models/visformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/visformer.py -------------------------------------------------------------------------------- /timm/models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/vision_transformer.py -------------------------------------------------------------------------------- /timm/models/vision_transformer_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/vision_transformer_hybrid.py -------------------------------------------------------------------------------- /timm/models/vovnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/vovnet.py -------------------------------------------------------------------------------- /timm/models/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/xception.py -------------------------------------------------------------------------------- /timm/models/xception_aligned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/xception_aligned.py -------------------------------------------------------------------------------- /timm/models/xcit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/models/xcit.py -------------------------------------------------------------------------------- /timm/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/__init__.py -------------------------------------------------------------------------------- /timm/optim/adabelief.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/adabelief.py -------------------------------------------------------------------------------- /timm/optim/adafactor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/adafactor.py -------------------------------------------------------------------------------- /timm/optim/adahessian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/adahessian.py -------------------------------------------------------------------------------- /timm/optim/adamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/adamp.py -------------------------------------------------------------------------------- /timm/optim/adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/adamw.py -------------------------------------------------------------------------------- /timm/optim/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/lookahead.py -------------------------------------------------------------------------------- /timm/optim/nadam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/nadam.py -------------------------------------------------------------------------------- /timm/optim/novograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/novograd.py -------------------------------------------------------------------------------- /timm/optim/nvnovograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/nvnovograd.py -------------------------------------------------------------------------------- /timm/optim/optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/optim_factory.py -------------------------------------------------------------------------------- /timm/optim/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/radam.py -------------------------------------------------------------------------------- /timm/optim/rmsprop_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/rmsprop_tf.py -------------------------------------------------------------------------------- /timm/optim/sgdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/optim/sgdp.py -------------------------------------------------------------------------------- /timm/scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/scheduler/__init__.py -------------------------------------------------------------------------------- /timm/scheduler/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/scheduler/cosine_lr.py -------------------------------------------------------------------------------- /timm/scheduler/multistep_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/scheduler/multistep_lr.py -------------------------------------------------------------------------------- /timm/scheduler/plateau_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/scheduler/plateau_lr.py -------------------------------------------------------------------------------- /timm/scheduler/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/scheduler/scheduler.py -------------------------------------------------------------------------------- /timm/scheduler/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/scheduler/scheduler_factory.py -------------------------------------------------------------------------------- /timm/scheduler/step_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/scheduler/step_lr.py -------------------------------------------------------------------------------- /timm/scheduler/tanh_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/scheduler/tanh_lr.py -------------------------------------------------------------------------------- /timm/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/__init__.py -------------------------------------------------------------------------------- /timm/utils/agc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/agc.py -------------------------------------------------------------------------------- /timm/utils/checkpoint_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/checkpoint_saver.py -------------------------------------------------------------------------------- /timm/utils/clip_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/clip_grad.py -------------------------------------------------------------------------------- /timm/utils/cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/cuda.py -------------------------------------------------------------------------------- /timm/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/distributed.py -------------------------------------------------------------------------------- /timm/utils/jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/jit.py -------------------------------------------------------------------------------- /timm/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/log.py -------------------------------------------------------------------------------- /timm/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/metrics.py -------------------------------------------------------------------------------- /timm/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/misc.py -------------------------------------------------------------------------------- /timm/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/model.py -------------------------------------------------------------------------------- /timm/utils/model_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/model_ema.py -------------------------------------------------------------------------------- /timm/utils/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/random.py -------------------------------------------------------------------------------- /timm/utils/summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/timm/utils/summary.py -------------------------------------------------------------------------------- /timm/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '0.4.13' 2 | -------------------------------------------------------------------------------- /visualizer/__init__.py: -------------------------------------------------------------------------------- 1 | from .visualizer import get_local -------------------------------------------------------------------------------- /visualizer/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luo3300612/Visualizer/HEAD/visualizer/visualizer.py --------------------------------------------------------------------------------