├── README.md ├── classic_models ├── __init__.py ├── alexnet.py ├── convnext.py ├── densenet.py ├── dla.py ├── efficientnet_v2.py ├── googlenet_v1.py ├── mlp_mixer.py ├── mobilenet_v1.py ├── mobilenet_v2.py ├── mobilenet_v3.py ├── repvgg.ipynb ├── resnet.py ├── shufflenet_v2.py ├── swin_transformer.py ├── van.py ├── vggnet.py ├── vision_transformer.py ├── xception.py └── zfnet.py ├── dataload ├── __pycache__ │ ├── dataload_five_flower.cpython-36.pyc │ ├── dataload_mini_imagenet.cpython-36.pyc │ └── dataload_mini_imagenet.cpython-39.pyc ├── dataload_five_flower.py └── dataload_mini_imagenet.py ├── test.py ├── train.py ├── train_distributed.py ├── train_sample.py └── utils ├── __init__.py ├── __pycache__ ├── __init__.cpython-36.pyc ├── __init__.cpython-39.pyc ├── distrubute_utils.cpython-36.pyc ├── distrubute_utils.cpython-39.pyc ├── lr_methods.cpython-36.pyc ├── lr_methods.cpython-39.pyc ├── train_engin.cpython-36.pyc └── train_engin.cpython-39.pyc ├── distrubute_utils.py ├── lr_methods.py ├── train_engin.py └── visualization.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/README.md -------------------------------------------------------------------------------- /classic_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/__init__.py -------------------------------------------------------------------------------- /classic_models/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/alexnet.py -------------------------------------------------------------------------------- /classic_models/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/convnext.py -------------------------------------------------------------------------------- /classic_models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/densenet.py -------------------------------------------------------------------------------- /classic_models/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/dla.py -------------------------------------------------------------------------------- /classic_models/efficientnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/efficientnet_v2.py -------------------------------------------------------------------------------- /classic_models/googlenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/googlenet_v1.py -------------------------------------------------------------------------------- /classic_models/mlp_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/mlp_mixer.py -------------------------------------------------------------------------------- /classic_models/mobilenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/mobilenet_v1.py -------------------------------------------------------------------------------- /classic_models/mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/mobilenet_v2.py -------------------------------------------------------------------------------- /classic_models/mobilenet_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/mobilenet_v3.py -------------------------------------------------------------------------------- /classic_models/repvgg.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/repvgg.ipynb -------------------------------------------------------------------------------- /classic_models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/resnet.py -------------------------------------------------------------------------------- /classic_models/shufflenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/shufflenet_v2.py -------------------------------------------------------------------------------- /classic_models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/swin_transformer.py -------------------------------------------------------------------------------- /classic_models/van.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/van.py -------------------------------------------------------------------------------- /classic_models/vggnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/vggnet.py -------------------------------------------------------------------------------- /classic_models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/vision_transformer.py -------------------------------------------------------------------------------- /classic_models/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/xception.py -------------------------------------------------------------------------------- /classic_models/zfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/classic_models/zfnet.py -------------------------------------------------------------------------------- /dataload/__pycache__/dataload_five_flower.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/dataload/__pycache__/dataload_five_flower.cpython-36.pyc -------------------------------------------------------------------------------- /dataload/__pycache__/dataload_mini_imagenet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/dataload/__pycache__/dataload_mini_imagenet.cpython-36.pyc -------------------------------------------------------------------------------- /dataload/__pycache__/dataload_mini_imagenet.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/dataload/__pycache__/dataload_mini_imagenet.cpython-39.pyc -------------------------------------------------------------------------------- /dataload/dataload_five_flower.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/dataload/dataload_five_flower.py -------------------------------------------------------------------------------- /dataload/dataload_mini_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/dataload/dataload_mini_imagenet.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/train.py -------------------------------------------------------------------------------- /train_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/train_distributed.py -------------------------------------------------------------------------------- /train_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/train_sample.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/utils/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/utils/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/distrubute_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/utils/__pycache__/distrubute_utils.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/distrubute_utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/utils/__pycache__/distrubute_utils.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/lr_methods.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/utils/__pycache__/lr_methods.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/lr_methods.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/utils/__pycache__/lr_methods.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/train_engin.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/utils/__pycache__/train_engin.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/train_engin.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/utils/__pycache__/train_engin.cpython-39.pyc -------------------------------------------------------------------------------- /utils/distrubute_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/utils/distrubute_utils.py -------------------------------------------------------------------------------- /utils/lr_methods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/utils/lr_methods.py -------------------------------------------------------------------------------- /utils/train_engin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/utils/train_engin.py -------------------------------------------------------------------------------- /utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arwin-Yu/Deep-Learning-Image-Classification-Models-Based-CNN-or-Attention/HEAD/utils/visualization.py --------------------------------------------------------------------------------