├── .gitignore ├── A_learning_notes ├── .ipynb_checkpoints │ ├── Q&A-checkpoint.ipynb │ └── generate_process-checkpoint.ipynb ├── Q&A.ipynb └── generate_process.ipynb ├── LICENSE ├── README.md ├── backbone ├── __init__.py ├── basic_backbone.py ├── mixnet18.py ├── mobilenet_v2.py ├── resnet18.py ├── resnet18_v2.py └── resnext.py ├── configs.py ├── dataset ├── __init__.py ├── dataset_util.py ├── file_util.py ├── test_sample │ ├── label.txt │ └── 鲁BC6T76.jpg └── tfrecord_util.py ├── images ├── GHM-insight.jpg ├── focal-loss.jpg ├── mixnet-18.svg ├── mobilenet-v2.svg ├── resnet-18-v2.svg ├── resnet-18.svg └── resnext-18.svg ├── multi_label ├── __init__.py ├── multi_label_loss.py ├── multi_label_model.py └── trainer.py ├── requirements.txt ├── run.py └── utils ├── __init__.py ├── check_label_file.py ├── draw_tools.py ├── logger_callback.py └── radam.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea/ 2 | **/*.py[cod] 3 | logs/ 4 | models/ -------------------------------------------------------------------------------- /A_learning_notes/.ipynb_checkpoints/Q&A-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/A_learning_notes/.ipynb_checkpoints/Q&A-checkpoint.ipynb -------------------------------------------------------------------------------- /A_learning_notes/.ipynb_checkpoints/generate_process-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/A_learning_notes/.ipynb_checkpoints/generate_process-checkpoint.ipynb -------------------------------------------------------------------------------- /A_learning_notes/Q&A.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/A_learning_notes/Q&A.ipynb -------------------------------------------------------------------------------- /A_learning_notes/generate_process.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/A_learning_notes/generate_process.ipynb -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/README.md -------------------------------------------------------------------------------- /backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/backbone/__init__.py -------------------------------------------------------------------------------- /backbone/basic_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/backbone/basic_backbone.py -------------------------------------------------------------------------------- /backbone/mixnet18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/backbone/mixnet18.py -------------------------------------------------------------------------------- /backbone/mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/backbone/mobilenet_v2.py -------------------------------------------------------------------------------- /backbone/resnet18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/backbone/resnet18.py -------------------------------------------------------------------------------- /backbone/resnet18_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/backbone/resnet18_v2.py -------------------------------------------------------------------------------- /backbone/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/backbone/resnext.py -------------------------------------------------------------------------------- /configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/configs.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/dataset_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/dataset/dataset_util.py -------------------------------------------------------------------------------- /dataset/file_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/dataset/file_util.py -------------------------------------------------------------------------------- /dataset/test_sample/label.txt: -------------------------------------------------------------------------------- 1 | 鲁BC6T76.jpg 15 1 12 6 27 7 6 -1 0 0 -------------------------------------------------------------------------------- /dataset/test_sample/鲁BC6T76.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/dataset/test_sample/鲁BC6T76.jpg -------------------------------------------------------------------------------- /dataset/tfrecord_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/dataset/tfrecord_util.py -------------------------------------------------------------------------------- /images/GHM-insight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/images/GHM-insight.jpg -------------------------------------------------------------------------------- /images/focal-loss.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/images/focal-loss.jpg -------------------------------------------------------------------------------- /images/mixnet-18.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/images/mixnet-18.svg -------------------------------------------------------------------------------- /images/mobilenet-v2.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/images/mobilenet-v2.svg -------------------------------------------------------------------------------- /images/resnet-18-v2.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/images/resnet-18-v2.svg -------------------------------------------------------------------------------- /images/resnet-18.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/images/resnet-18.svg -------------------------------------------------------------------------------- /images/resnext-18.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/images/resnext-18.svg -------------------------------------------------------------------------------- /multi_label/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/multi_label/__init__.py -------------------------------------------------------------------------------- /multi_label/multi_label_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/multi_label/multi_label_loss.py -------------------------------------------------------------------------------- /multi_label/multi_label_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/multi_label/multi_label_model.py -------------------------------------------------------------------------------- /multi_label/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/multi_label/trainer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/run.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/check_label_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/utils/check_label_file.py -------------------------------------------------------------------------------- /utils/draw_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/utils/draw_tools.py -------------------------------------------------------------------------------- /utils/logger_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/utils/logger_callback.py -------------------------------------------------------------------------------- /utils/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zheng-yuwei/multi-label-classification/HEAD/utils/radam.py --------------------------------------------------------------------------------