├── LICENSE ├── README.md ├── cls_models ├── __init__.py └── model.py ├── cpp_inference ├── .vscode │ └── settings.json ├── CMakeLists.txt ├── classification │ ├── CMakeLists.txt │ ├── include │ │ └── ImgCls.hpp │ ├── src │ │ └── ImgCls.cpp │ └── test_ImgCls.cpp ├── compile.sh └── traced_model │ └── trace_model.py ├── dataset ├── __init__.py ├── cls_dataset.py └── transform.py ├── ensamble ├── kaggle_vote.py └── samples │ ├── method_1.csv │ ├── method_2.csv │ └── method_3.csv ├── run.sh ├── sample_files └── imgs │ ├── cat │ ├── 0.jpg │ ├── 1.jpg │ ├── 2.jpg │ ├── 3.jpg │ ├── 4.jpg │ └── 5.jpg │ ├── dog │ ├── 0.jpg │ ├── 1.jpg │ ├── 2.jpg │ ├── 3.jpg │ ├── 4.jpg │ └── 5.jpg │ └── listfile.txt ├── tools ├── evaluation.py ├── predict.py └── train_val.py ├── trt_inference ├── CMakeLists.txt ├── convert_onnx.py ├── include │ └── TrtCls.hpp ├── labels.txt └── src │ └── TrtCls.cpp ├── utils ├── __init__.py ├── label_smoothing_pytorch.py ├── loss_kd.py ├── lr_scheduler.py ├── opts.py ├── util.py └── warmup_lr.py └── visualization ├── Feature_Visualization.ipynb ├── Feature_Visualization.py ├── f1_conv1.png └── test.png /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/README.md -------------------------------------------------------------------------------- /cls_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/cls_models/__init__.py -------------------------------------------------------------------------------- /cls_models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/cls_models/model.py -------------------------------------------------------------------------------- /cpp_inference/.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/cpp_inference/.vscode/settings.json -------------------------------------------------------------------------------- /cpp_inference/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/cpp_inference/CMakeLists.txt -------------------------------------------------------------------------------- /cpp_inference/classification/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/cpp_inference/classification/CMakeLists.txt -------------------------------------------------------------------------------- /cpp_inference/classification/include/ImgCls.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/cpp_inference/classification/include/ImgCls.hpp -------------------------------------------------------------------------------- /cpp_inference/classification/src/ImgCls.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/cpp_inference/classification/src/ImgCls.cpp -------------------------------------------------------------------------------- /cpp_inference/classification/test_ImgCls.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/cpp_inference/classification/test_ImgCls.cpp -------------------------------------------------------------------------------- /cpp_inference/compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/cpp_inference/compile.sh -------------------------------------------------------------------------------- /cpp_inference/traced_model/trace_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/cpp_inference/traced_model/trace_model.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/cls_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/dataset/cls_dataset.py -------------------------------------------------------------------------------- /dataset/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/dataset/transform.py -------------------------------------------------------------------------------- /ensamble/kaggle_vote.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/ensamble/kaggle_vote.py -------------------------------------------------------------------------------- /ensamble/samples/method_1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/ensamble/samples/method_1.csv -------------------------------------------------------------------------------- /ensamble/samples/method_2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/ensamble/samples/method_2.csv -------------------------------------------------------------------------------- /ensamble/samples/method_3.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/ensamble/samples/method_3.csv -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/run.sh -------------------------------------------------------------------------------- /sample_files/imgs/cat/0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/cat/0.jpg -------------------------------------------------------------------------------- /sample_files/imgs/cat/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/cat/1.jpg -------------------------------------------------------------------------------- /sample_files/imgs/cat/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/cat/2.jpg -------------------------------------------------------------------------------- /sample_files/imgs/cat/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/cat/3.jpg -------------------------------------------------------------------------------- /sample_files/imgs/cat/4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/cat/4.jpg -------------------------------------------------------------------------------- /sample_files/imgs/cat/5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/cat/5.jpg -------------------------------------------------------------------------------- /sample_files/imgs/dog/0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/dog/0.jpg -------------------------------------------------------------------------------- /sample_files/imgs/dog/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/dog/1.jpg -------------------------------------------------------------------------------- /sample_files/imgs/dog/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/dog/2.jpg -------------------------------------------------------------------------------- /sample_files/imgs/dog/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/dog/3.jpg -------------------------------------------------------------------------------- /sample_files/imgs/dog/4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/dog/4.jpg -------------------------------------------------------------------------------- /sample_files/imgs/dog/5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/dog/5.jpg -------------------------------------------------------------------------------- /sample_files/imgs/listfile.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/sample_files/imgs/listfile.txt -------------------------------------------------------------------------------- /tools/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/tools/evaluation.py -------------------------------------------------------------------------------- /tools/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/tools/predict.py -------------------------------------------------------------------------------- /tools/train_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/tools/train_val.py -------------------------------------------------------------------------------- /trt_inference/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/trt_inference/CMakeLists.txt -------------------------------------------------------------------------------- /trt_inference/convert_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/trt_inference/convert_onnx.py -------------------------------------------------------------------------------- /trt_inference/include/TrtCls.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/trt_inference/include/TrtCls.hpp -------------------------------------------------------------------------------- /trt_inference/labels.txt: -------------------------------------------------------------------------------- 1 | 0 cat 2 | 1 dog 3 | -------------------------------------------------------------------------------- /trt_inference/src/TrtCls.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/trt_inference/src/TrtCls.cpp -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/label_smoothing_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/utils/label_smoothing_pytorch.py -------------------------------------------------------------------------------- /utils/loss_kd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/utils/loss_kd.py -------------------------------------------------------------------------------- /utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/utils/lr_scheduler.py -------------------------------------------------------------------------------- /utils/opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/utils/opts.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/utils/util.py -------------------------------------------------------------------------------- /utils/warmup_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/utils/warmup_lr.py -------------------------------------------------------------------------------- /visualization/Feature_Visualization.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/visualization/Feature_Visualization.ipynb -------------------------------------------------------------------------------- /visualization/Feature_Visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/visualization/Feature_Visualization.py -------------------------------------------------------------------------------- /visualization/f1_conv1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/visualization/f1_conv1.png -------------------------------------------------------------------------------- /visualization/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxztju/pytorch_classification/HEAD/visualization/test.png --------------------------------------------------------------------------------