├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── Teacher_Inference.py ├── datasets └── README.md ├── demo ├── MegEngine │ ├── cpp │ │ ├── README.md │ │ ├── build.sh │ │ └── yolox.cpp │ └── python │ │ ├── README.md │ │ ├── build.py │ │ ├── convert_weights.py │ │ ├── demo.py │ │ ├── dump.py │ │ └── models │ │ ├── __init__.py │ │ ├── darknet.py │ │ ├── network_blocks.py │ │ ├── yolo_fpn.py │ │ ├── yolo_head.py │ │ ├── yolo_pafpn.py │ │ └── yolox.py ├── ONNXRuntime │ ├── README.md │ └── onnx_inference.py ├── OpenVINO │ ├── README.md │ ├── cpp │ │ ├── CMakeLists.txt │ │ ├── README.md │ │ └── yolox_openvino.cpp │ └── python │ │ ├── README.md │ │ └── openvino_inference.py ├── TensorRT │ ├── cpp │ │ ├── CMakeLists.txt │ │ ├── README.md │ │ ├── logging.h │ │ └── yolox.cpp │ └── python │ │ └── README.md ├── ncnn │ ├── README.md │ ├── android │ │ ├── README.md │ │ ├── app │ │ │ ├── build.gradle │ │ │ └── src │ │ │ │ └── main │ │ │ │ ├── AndroidManifest.xml │ │ │ │ ├── assets │ │ │ │ └── yolox.param │ │ │ │ ├── java │ │ │ │ └── com │ │ │ │ │ └── megvii │ │ │ │ │ └── yoloXncnn │ │ │ │ │ ├── MainActivity.java │ │ │ │ │ ├── YOLOXncnn.java │ │ │ │ │ └── yoloXncnn.java │ │ │ │ ├── jni │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── yoloXncnn_jni.cpp │ │ │ │ └── res │ │ │ │ ├── layout │ │ │ │ └── main.xml │ │ │ │ └── values │ │ │ │ └── strings.xml │ │ ├── build.gradle │ │ ├── gradle │ │ │ └── wrapper │ │ │ │ ├── gradle-wrapper.jar │ │ │ │ └── gradle-wrapper.properties │ │ ├── gradlew │ │ ├── gradlew.bat │ │ └── settings.gradle │ └── cpp │ │ ├── README.md │ │ └── yolox.cpp └── nebullvm │ ├── README.md │ └── nebullvm_optimization.py ├── docs ├── .gitignore ├── Makefile ├── _static │ └── css │ │ └── custom.css ├── assignment_visualization.md ├── cache.md ├── conf.py ├── demo │ ├── megengine_cpp_readme.md │ ├── megengine_py_readme.md │ ├── ncnn_android_readme.md │ ├── ncnn_cpp_readme.md │ ├── onnx_readme.md │ ├── openvino_cpp_readme.md │ ├── openvino_py_readme.md │ ├── trt_cpp_readme.md │ └── trt_py_readme.md ├── freeze_module.md ├── index.rst ├── manipulate_training_image_size.md ├── model_zoo.md ├── quick_run.md ├── requirements-doc.txt ├── train_custom_data.md └── updates_note.md ├── exps ├── default │ ├── __init__.py │ ├── yolov3.py │ ├── yolox_l.py │ ├── yolox_m.py │ ├── yolox_nano.py │ ├── yolox_s.py │ ├── yolox_tiny.py │ └── yolox_x.py └── example │ ├── custom │ ├── nano.py │ └── yolox_s.py │ └── yolox_voc │ └── yolox_voc_s.py ├── hubconf.py ├── images ├── Comparison-YOLOX-YOLOX-ViT.png ├── KD-Comparison.png ├── KD-github.png ├── Launch-ViT.png ├── Offline-KD.png ├── Online-KD-Code.png ├── Online-KD.png ├── YoloX-ViT-Model.png ├── eu-commission.jpg └── remaro-right-1024.png ├── requirements.txt ├── setup.cfg ├── setup.py ├── tests ├── __init__.py └── utils │ └── test_model_utils.py ├── tools ├── __init__.py ├── demo.py ├── eval.py ├── export_onnx.py ├── export_torchscript.py ├── train.py ├── trt.py └── visualize_assign.py └── yolox ├── __init__.py ├── core ├── __init__.py ├── launch.py └── trainer.py ├── data ├── __init__.py ├── data_augment.py ├── data_prefetcher.py ├── dataloading.py ├── datasets │ ├── __init__.py │ ├── coco.py │ ├── coco_classes.py │ ├── datasets_wrapper.py │ ├── mosaicdetection.py │ ├── voc.py │ └── voc_classes.py └── samplers.py ├── evaluators ├── __init__.py ├── coco_evaluator.py ├── voc_eval.py └── voc_evaluator.py ├── exp ├── __init__.py ├── base_exp.py ├── build.py ├── default │ └── __init__.py └── yolox_base.py ├── layers ├── __init__.py ├── cocoeval │ ├── cocoeval.cpp │ └── cocoeval.h ├── fast_coco_eval_api.py └── jit_ops.py ├── models ├── __init__.py ├── build.py ├── darknet.py ├── losses.py ├── network_blocks.py ├── yolo_fpn.py ├── yolo_head.py ├── yolo_pafpn.py └── yolox.py ├── tools └── __init__.py └── utils ├── __init__.py ├── allreduce_norm.py ├── boxes.py ├── checkpoint.py ├── compat.py ├── demo_utils.py ├── dist.py ├── ema.py ├── logger.py ├── lr_scheduler.py ├── metric.py ├── model_utils.py ├── setup_env.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/README.md -------------------------------------------------------------------------------- /Teacher_Inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/Teacher_Inference.py -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/datasets/README.md -------------------------------------------------------------------------------- /demo/MegEngine/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/cpp/README.md -------------------------------------------------------------------------------- /demo/MegEngine/cpp/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/cpp/build.sh -------------------------------------------------------------------------------- /demo/MegEngine/cpp/yolox.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/cpp/yolox.cpp -------------------------------------------------------------------------------- /demo/MegEngine/python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/python/README.md -------------------------------------------------------------------------------- /demo/MegEngine/python/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/python/build.py -------------------------------------------------------------------------------- /demo/MegEngine/python/convert_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/python/convert_weights.py -------------------------------------------------------------------------------- /demo/MegEngine/python/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/python/demo.py -------------------------------------------------------------------------------- /demo/MegEngine/python/dump.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/python/dump.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/python/models/__init__.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/python/models/darknet.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/network_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/python/models/network_blocks.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/yolo_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/python/models/yolo_fpn.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/yolo_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/python/models/yolo_head.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/yolo_pafpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/python/models/yolo_pafpn.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/yolox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/MegEngine/python/models/yolox.py -------------------------------------------------------------------------------- /demo/ONNXRuntime/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ONNXRuntime/README.md -------------------------------------------------------------------------------- /demo/ONNXRuntime/onnx_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ONNXRuntime/onnx_inference.py -------------------------------------------------------------------------------- /demo/OpenVINO/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/OpenVINO/README.md -------------------------------------------------------------------------------- /demo/OpenVINO/cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/OpenVINO/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /demo/OpenVINO/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/OpenVINO/cpp/README.md -------------------------------------------------------------------------------- /demo/OpenVINO/cpp/yolox_openvino.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/OpenVINO/cpp/yolox_openvino.cpp -------------------------------------------------------------------------------- /demo/OpenVINO/python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/OpenVINO/python/README.md -------------------------------------------------------------------------------- /demo/OpenVINO/python/openvino_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/OpenVINO/python/openvino_inference.py -------------------------------------------------------------------------------- /demo/TensorRT/cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/TensorRT/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /demo/TensorRT/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/TensorRT/cpp/README.md -------------------------------------------------------------------------------- /demo/TensorRT/cpp/logging.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/TensorRT/cpp/logging.h -------------------------------------------------------------------------------- /demo/TensorRT/cpp/yolox.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/TensorRT/cpp/yolox.cpp -------------------------------------------------------------------------------- /demo/TensorRT/python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/TensorRT/python/README.md -------------------------------------------------------------------------------- /demo/ncnn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/README.md -------------------------------------------------------------------------------- /demo/ncnn/android/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/README.md -------------------------------------------------------------------------------- /demo/ncnn/android/app/build.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/app/build.gradle -------------------------------------------------------------------------------- /demo/ncnn/android/app/src/main/AndroidManifest.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/app/src/main/AndroidManifest.xml -------------------------------------------------------------------------------- /demo/ncnn/android/app/src/main/assets/yolox.param: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/app/src/main/assets/yolox.param -------------------------------------------------------------------------------- /demo/ncnn/android/app/src/main/java/com/megvii/yoloXncnn/MainActivity.java: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/app/src/main/java/com/megvii/yoloXncnn/MainActivity.java -------------------------------------------------------------------------------- /demo/ncnn/android/app/src/main/java/com/megvii/yoloXncnn/YOLOXncnn.java: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/app/src/main/java/com/megvii/yoloXncnn/YOLOXncnn.java -------------------------------------------------------------------------------- /demo/ncnn/android/app/src/main/java/com/megvii/yoloXncnn/yoloXncnn.java: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/app/src/main/java/com/megvii/yoloXncnn/yoloXncnn.java -------------------------------------------------------------------------------- /demo/ncnn/android/app/src/main/jni/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/app/src/main/jni/CMakeLists.txt -------------------------------------------------------------------------------- /demo/ncnn/android/app/src/main/jni/yoloXncnn_jni.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/app/src/main/jni/yoloXncnn_jni.cpp -------------------------------------------------------------------------------- /demo/ncnn/android/app/src/main/res/layout/main.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/app/src/main/res/layout/main.xml -------------------------------------------------------------------------------- /demo/ncnn/android/app/src/main/res/values/strings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/app/src/main/res/values/strings.xml -------------------------------------------------------------------------------- /demo/ncnn/android/build.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/build.gradle -------------------------------------------------------------------------------- /demo/ncnn/android/gradle/wrapper/gradle-wrapper.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/gradle/wrapper/gradle-wrapper.jar -------------------------------------------------------------------------------- /demo/ncnn/android/gradle/wrapper/gradle-wrapper.properties: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/gradle/wrapper/gradle-wrapper.properties -------------------------------------------------------------------------------- /demo/ncnn/android/gradlew: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/gradlew -------------------------------------------------------------------------------- /demo/ncnn/android/gradlew.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/android/gradlew.bat -------------------------------------------------------------------------------- /demo/ncnn/android/settings.gradle: -------------------------------------------------------------------------------- 1 | include ':app' 2 | -------------------------------------------------------------------------------- /demo/ncnn/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/cpp/README.md -------------------------------------------------------------------------------- /demo/ncnn/cpp/yolox.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/ncnn/cpp/yolox.cpp -------------------------------------------------------------------------------- /demo/nebullvm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/nebullvm/README.md -------------------------------------------------------------------------------- /demo/nebullvm/nebullvm_optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/demo/nebullvm/nebullvm_optimization.py -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- 1 | _build -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_static/css/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/_static/css/custom.css -------------------------------------------------------------------------------- /docs/assignment_visualization.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/assignment_visualization.md -------------------------------------------------------------------------------- /docs/cache.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/cache.md -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/demo/megengine_cpp_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/MegEngine/cpp/README.md -------------------------------------------------------------------------------- /docs/demo/megengine_py_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/MegEngine/python/README.md -------------------------------------------------------------------------------- /docs/demo/ncnn_android_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/ncnn/android/README.md -------------------------------------------------------------------------------- /docs/demo/ncnn_cpp_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/ncnn/cpp/README.md -------------------------------------------------------------------------------- /docs/demo/onnx_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/ONNXRuntime/README.md -------------------------------------------------------------------------------- /docs/demo/openvino_cpp_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/OpenVINO/cpp/README.md -------------------------------------------------------------------------------- /docs/demo/openvino_py_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/OpenVINO/python/README.md -------------------------------------------------------------------------------- /docs/demo/trt_cpp_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/TensorRT/cpp/README.md -------------------------------------------------------------------------------- /docs/demo/trt_py_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/TensorRT/python/README.md -------------------------------------------------------------------------------- /docs/freeze_module.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/freeze_module.md -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/manipulate_training_image_size.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/manipulate_training_image_size.md -------------------------------------------------------------------------------- /docs/model_zoo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/model_zoo.md -------------------------------------------------------------------------------- /docs/quick_run.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/quick_run.md -------------------------------------------------------------------------------- /docs/requirements-doc.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/requirements-doc.txt -------------------------------------------------------------------------------- /docs/train_custom_data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/train_custom_data.md -------------------------------------------------------------------------------- /docs/updates_note.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/docs/updates_note.md -------------------------------------------------------------------------------- /exps/default/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/exps/default/__init__.py -------------------------------------------------------------------------------- /exps/default/yolov3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/exps/default/yolov3.py -------------------------------------------------------------------------------- /exps/default/yolox_l.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/exps/default/yolox_l.py -------------------------------------------------------------------------------- /exps/default/yolox_m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/exps/default/yolox_m.py -------------------------------------------------------------------------------- /exps/default/yolox_nano.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/exps/default/yolox_nano.py -------------------------------------------------------------------------------- /exps/default/yolox_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/exps/default/yolox_s.py -------------------------------------------------------------------------------- /exps/default/yolox_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/exps/default/yolox_tiny.py -------------------------------------------------------------------------------- /exps/default/yolox_x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/exps/default/yolox_x.py -------------------------------------------------------------------------------- /exps/example/custom/nano.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/exps/example/custom/nano.py -------------------------------------------------------------------------------- /exps/example/custom/yolox_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/exps/example/custom/yolox_s.py -------------------------------------------------------------------------------- /exps/example/yolox_voc/yolox_voc_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/exps/example/yolox_voc/yolox_voc_s.py -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/hubconf.py -------------------------------------------------------------------------------- /images/Comparison-YOLOX-YOLOX-ViT.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/images/Comparison-YOLOX-YOLOX-ViT.png -------------------------------------------------------------------------------- /images/KD-Comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/images/KD-Comparison.png -------------------------------------------------------------------------------- /images/KD-github.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/images/KD-github.png -------------------------------------------------------------------------------- /images/Launch-ViT.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/images/Launch-ViT.png -------------------------------------------------------------------------------- /images/Offline-KD.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/images/Offline-KD.png -------------------------------------------------------------------------------- /images/Online-KD-Code.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/images/Online-KD-Code.png -------------------------------------------------------------------------------- /images/Online-KD.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/images/Online-KD.png -------------------------------------------------------------------------------- /images/YoloX-ViT-Model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/images/YoloX-ViT-Model.png -------------------------------------------------------------------------------- /images/eu-commission.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/images/eu-commission.jpg -------------------------------------------------------------------------------- /images/remaro-right-1024.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/images/remaro-right-1024.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python3 2 | # -*- coding:utf-8 -*- 3 | -------------------------------------------------------------------------------- /tests/utils/test_model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/tests/utils/test_model_utils.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/tools/__init__.py -------------------------------------------------------------------------------- /tools/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/tools/demo.py -------------------------------------------------------------------------------- /tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/tools/eval.py -------------------------------------------------------------------------------- /tools/export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/tools/export_onnx.py -------------------------------------------------------------------------------- /tools/export_torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/tools/export_torchscript.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/tools/trt.py -------------------------------------------------------------------------------- /tools/visualize_assign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/tools/visualize_assign.py -------------------------------------------------------------------------------- /yolox/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python3 2 | # -*- coding:utf-8 -*- 3 | 4 | __version__ = "0.3.0" 5 | -------------------------------------------------------------------------------- /yolox/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/core/__init__.py -------------------------------------------------------------------------------- /yolox/core/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/core/launch.py -------------------------------------------------------------------------------- /yolox/core/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/core/trainer.py -------------------------------------------------------------------------------- /yolox/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/data/__init__.py -------------------------------------------------------------------------------- /yolox/data/data_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/data/data_augment.py -------------------------------------------------------------------------------- /yolox/data/data_prefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/data/data_prefetcher.py -------------------------------------------------------------------------------- /yolox/data/dataloading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/data/dataloading.py -------------------------------------------------------------------------------- /yolox/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/data/datasets/__init__.py -------------------------------------------------------------------------------- /yolox/data/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/data/datasets/coco.py -------------------------------------------------------------------------------- /yolox/data/datasets/coco_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/data/datasets/coco_classes.py -------------------------------------------------------------------------------- /yolox/data/datasets/datasets_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/data/datasets/datasets_wrapper.py -------------------------------------------------------------------------------- /yolox/data/datasets/mosaicdetection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/data/datasets/mosaicdetection.py -------------------------------------------------------------------------------- /yolox/data/datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/data/datasets/voc.py -------------------------------------------------------------------------------- /yolox/data/datasets/voc_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/data/datasets/voc_classes.py -------------------------------------------------------------------------------- /yolox/data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/data/samplers.py -------------------------------------------------------------------------------- /yolox/evaluators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/evaluators/__init__.py -------------------------------------------------------------------------------- /yolox/evaluators/coco_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/evaluators/coco_evaluator.py -------------------------------------------------------------------------------- /yolox/evaluators/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/evaluators/voc_eval.py -------------------------------------------------------------------------------- /yolox/evaluators/voc_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/evaluators/voc_evaluator.py -------------------------------------------------------------------------------- /yolox/exp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/exp/__init__.py -------------------------------------------------------------------------------- /yolox/exp/base_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/exp/base_exp.py -------------------------------------------------------------------------------- /yolox/exp/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/exp/build.py -------------------------------------------------------------------------------- /yolox/exp/default/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/exp/default/__init__.py -------------------------------------------------------------------------------- /yolox/exp/yolox_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/exp/yolox_base.py -------------------------------------------------------------------------------- /yolox/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/layers/__init__.py -------------------------------------------------------------------------------- /yolox/layers/cocoeval/cocoeval.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/layers/cocoeval/cocoeval.cpp -------------------------------------------------------------------------------- /yolox/layers/cocoeval/cocoeval.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/layers/cocoeval/cocoeval.h -------------------------------------------------------------------------------- /yolox/layers/fast_coco_eval_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/layers/fast_coco_eval_api.py -------------------------------------------------------------------------------- /yolox/layers/jit_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/layers/jit_ops.py -------------------------------------------------------------------------------- /yolox/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/models/__init__.py -------------------------------------------------------------------------------- /yolox/models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/models/build.py -------------------------------------------------------------------------------- /yolox/models/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/models/darknet.py -------------------------------------------------------------------------------- /yolox/models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/models/losses.py -------------------------------------------------------------------------------- /yolox/models/network_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/models/network_blocks.py -------------------------------------------------------------------------------- /yolox/models/yolo_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/models/yolo_fpn.py -------------------------------------------------------------------------------- /yolox/models/yolo_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/models/yolo_head.py -------------------------------------------------------------------------------- /yolox/models/yolo_pafpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/models/yolo_pafpn.py -------------------------------------------------------------------------------- /yolox/models/yolox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/models/yolox.py -------------------------------------------------------------------------------- /yolox/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/tools/__init__.py -------------------------------------------------------------------------------- /yolox/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/__init__.py -------------------------------------------------------------------------------- /yolox/utils/allreduce_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/allreduce_norm.py -------------------------------------------------------------------------------- /yolox/utils/boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/boxes.py -------------------------------------------------------------------------------- /yolox/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/checkpoint.py -------------------------------------------------------------------------------- /yolox/utils/compat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/compat.py -------------------------------------------------------------------------------- /yolox/utils/demo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/demo_utils.py -------------------------------------------------------------------------------- /yolox/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/dist.py -------------------------------------------------------------------------------- /yolox/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/ema.py -------------------------------------------------------------------------------- /yolox/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/logger.py -------------------------------------------------------------------------------- /yolox/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/lr_scheduler.py -------------------------------------------------------------------------------- /yolox/utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/metric.py -------------------------------------------------------------------------------- /yolox/utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/model_utils.py -------------------------------------------------------------------------------- /yolox/utils/setup_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/setup_env.py -------------------------------------------------------------------------------- /yolox/utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/remaro-network/KD-YOLOX-ViT/HEAD/yolox/utils/visualize.py --------------------------------------------------------------------------------