├── .gitignore ├── LICENSE ├── README.md ├── classification.py ├── datasets ├── test │ └── README.md └── train │ └── README.md ├── eval.py ├── img ├── cat.jpg └── dog.jpg ├── logs └── README.md ├── model_data ├── cls_classes.txt ├── mobilenet025_catvsdog.h5 └── mobilenet_2_5_224_tf_no_top.h5 ├── nets ├── __init__.py ├── mobilenetv1.py ├── mobilenetv2.py ├── resnet.py ├── swin_transformer.py ├── vgg.py └── vision_transformer.py ├── predict.py ├── requirements.txt ├── summary.py ├── train.py ├── txt_annotation.py └── utils ├── __init__.py ├── callbacks.py ├── dataloader.py ├── utils.py ├── utils_aug.py └── utils_metrics.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/README.md -------------------------------------------------------------------------------- /classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/classification.py -------------------------------------------------------------------------------- /datasets/test/README.md: -------------------------------------------------------------------------------- 1 | 用于存放测试图片 -------------------------------------------------------------------------------- /datasets/train/README.md: -------------------------------------------------------------------------------- 1 | 用于存放训练图片 -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/eval.py -------------------------------------------------------------------------------- /img/cat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/img/cat.jpg -------------------------------------------------------------------------------- /img/dog.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/img/dog.jpg -------------------------------------------------------------------------------- /logs/README.md: -------------------------------------------------------------------------------- 1 | 存放训练后的模型 -------------------------------------------------------------------------------- /model_data/cls_classes.txt: -------------------------------------------------------------------------------- 1 | cat 2 | dog -------------------------------------------------------------------------------- /model_data/mobilenet025_catvsdog.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/model_data/mobilenet025_catvsdog.h5 -------------------------------------------------------------------------------- /model_data/mobilenet_2_5_224_tf_no_top.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/model_data/mobilenet_2_5_224_tf_no_top.h5 -------------------------------------------------------------------------------- /nets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/nets/__init__.py -------------------------------------------------------------------------------- /nets/mobilenetv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/nets/mobilenetv1.py -------------------------------------------------------------------------------- /nets/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/nets/mobilenetv2.py -------------------------------------------------------------------------------- /nets/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/nets/resnet.py -------------------------------------------------------------------------------- /nets/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/nets/swin_transformer.py -------------------------------------------------------------------------------- /nets/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/nets/vgg.py -------------------------------------------------------------------------------- /nets/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/nets/vision_transformer.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/predict.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/requirements.txt -------------------------------------------------------------------------------- /summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/summary.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/train.py -------------------------------------------------------------------------------- /txt_annotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/txt_annotation.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/utils/callbacks.py -------------------------------------------------------------------------------- /utils/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/utils/dataloader.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/utils/utils.py -------------------------------------------------------------------------------- /utils/utils_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/utils/utils_aug.py -------------------------------------------------------------------------------- /utils/utils_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bubbliiiing/classification-keras/HEAD/utils/utils_metrics.py --------------------------------------------------------------------------------