├── .gitignore ├── .readthedocs.yaml ├── 123.py ├── LICENSE ├── README.md ├── Untitled.ipynb ├── assets ├── demo.png ├── dog.jpg ├── git_fig.png └── logo.png ├── datasets ├── README.md ├── ccpd │ ├── voctococo.py │ └── yolotovoc.py ├── plate_kp │ ├── voctococo.py │ └── yolotovoc.py └── plate_seg │ ├── 50rle.zip │ ├── voctococo.py │ └── yolotovoc.py ├── 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 │ ├── android │ ├── README.md │ ├── app │ │ ├── build.gradle │ │ └── src │ │ │ └── main │ │ │ ├── AndroidManifest.xml │ │ │ ├── assets │ │ │ └── yolox.param │ │ │ ├── java │ │ │ └── com │ │ │ │ └── megvii │ │ │ │ └── yoloXncnn │ │ │ │ ├── MainActivity.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 ├── docs ├── .gitignore ├── Makefile ├── _static │ └── css │ │ └── custom.css ├── 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 ├── 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 │ ├── nano.py │ ├── yolov3.py │ ├── yolox_ccpd.py │ ├── yolox_decorate.py │ ├── yolox_kp.py │ ├── yolox_l.py │ ├── yolox_m.py │ ├── yolox_masonry.py │ ├── yolox_plaster.py │ ├── yolox_s.py │ ├── yolox_seg.py │ ├── yolox_tiny.py │ └── yolox_x.py └── example │ ├── custom │ ├── nano.py │ └── yolox_s.py │ └── yolox_voc │ └── yolox_voc_s.py ├── imgs ├── kps │ ├── 5.jpg │ ├── 6.jpg │ └── 7.jpg └── segs │ ├── 1.jpg │ ├── 2.jpg │ ├── 3.jpg │ ├── 4.jpg │ └── 5.jpg ├── my_trt.py ├── requirements.txt ├── setup.cfg ├── setup.py ├── tests ├── __init__.py └── utils │ └── test_model_utils.py ├── tools ├── __init__.py ├── demo.py ├── demo_panoptic.py ├── eval.py ├── export_onnx.py ├── export_torchscript.py ├── train.py └── trt.py ├── trt_convert.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 │ │ ├── generate_wall_data.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 │ └── yolox_base.py ├── layers │ ├── __init__.py │ ├── fast_coco_eval_api.py │ └── jit_ops.py ├── models │ ├── __init__.py │ ├── coatnet.py │ ├── darknet.py │ ├── losses.py │ ├── network_blocks.py │ ├── yolo_fpn.py │ ├── yolo_head.py │ ├── yolo_head_panoptic.py │ ├── yolo_pafpn.py │ ├── yolo_pafpn_slim.py │ ├── yolov7_tiny.py │ └── yolox.py └── utils │ ├── __init__.py │ ├── allreduce_norm.py │ ├── boxes.py │ ├── checkpoint.py │ ├── compat.py │ ├── demo_utils.py │ ├── dist.py │ ├── ema.py │ ├── fuse_model.py │ ├── logger.py │ ├── lr_scheduler.py │ ├── metric.py │ ├── model_utils.py │ ├── setup_env.py │ └── visualize.py └── 注意.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/.gitignore -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /123.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/123.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/README.md -------------------------------------------------------------------------------- /Untitled.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/Untitled.ipynb -------------------------------------------------------------------------------- /assets/demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/assets/demo.png -------------------------------------------------------------------------------- /assets/dog.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/assets/dog.jpg -------------------------------------------------------------------------------- /assets/git_fig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/assets/git_fig.png -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/assets/logo.png -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/datasets/README.md -------------------------------------------------------------------------------- /datasets/ccpd/voctococo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/datasets/ccpd/voctococo.py -------------------------------------------------------------------------------- /datasets/ccpd/yolotovoc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/datasets/ccpd/yolotovoc.py -------------------------------------------------------------------------------- /datasets/plate_kp/voctococo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/datasets/plate_kp/voctococo.py -------------------------------------------------------------------------------- /datasets/plate_kp/yolotovoc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/datasets/plate_kp/yolotovoc.py -------------------------------------------------------------------------------- /datasets/plate_seg/50rle.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/datasets/plate_seg/50rle.zip -------------------------------------------------------------------------------- /datasets/plate_seg/voctococo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/datasets/plate_seg/voctococo.py -------------------------------------------------------------------------------- /datasets/plate_seg/yolotovoc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/datasets/plate_seg/yolotovoc.py -------------------------------------------------------------------------------- /demo/MegEngine/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/cpp/README.md -------------------------------------------------------------------------------- /demo/MegEngine/cpp/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/cpp/build.sh -------------------------------------------------------------------------------- /demo/MegEngine/cpp/yolox.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/cpp/yolox.cpp -------------------------------------------------------------------------------- /demo/MegEngine/python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/python/README.md -------------------------------------------------------------------------------- /demo/MegEngine/python/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/python/build.py -------------------------------------------------------------------------------- /demo/MegEngine/python/convert_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/python/convert_weights.py -------------------------------------------------------------------------------- /demo/MegEngine/python/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/python/demo.py -------------------------------------------------------------------------------- /demo/MegEngine/python/dump.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/python/dump.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/python/models/__init__.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/python/models/darknet.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/network_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/python/models/network_blocks.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/yolo_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/python/models/yolo_fpn.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/yolo_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/python/models/yolo_head.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/yolo_pafpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/python/models/yolo_pafpn.py -------------------------------------------------------------------------------- /demo/MegEngine/python/models/yolox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/MegEngine/python/models/yolox.py -------------------------------------------------------------------------------- /demo/ONNXRuntime/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/ONNXRuntime/README.md -------------------------------------------------------------------------------- /demo/ONNXRuntime/onnx_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/ONNXRuntime/onnx_inference.py -------------------------------------------------------------------------------- /demo/OpenVINO/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/OpenVINO/README.md -------------------------------------------------------------------------------- /demo/OpenVINO/cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/OpenVINO/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /demo/OpenVINO/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/OpenVINO/cpp/README.md -------------------------------------------------------------------------------- /demo/OpenVINO/cpp/yolox_openvino.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/OpenVINO/cpp/yolox_openvino.cpp -------------------------------------------------------------------------------- /demo/OpenVINO/python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/OpenVINO/python/README.md -------------------------------------------------------------------------------- /demo/OpenVINO/python/openvino_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/OpenVINO/python/openvino_inference.py -------------------------------------------------------------------------------- /demo/TensorRT/cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/TensorRT/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /demo/TensorRT/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/TensorRT/cpp/README.md -------------------------------------------------------------------------------- /demo/TensorRT/cpp/logging.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/TensorRT/cpp/logging.h -------------------------------------------------------------------------------- /demo/TensorRT/cpp/yolox.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/TensorRT/cpp/yolox.cpp -------------------------------------------------------------------------------- /demo/TensorRT/python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/TensorRT/python/README.md -------------------------------------------------------------------------------- /demo/ncnn/android/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/ncnn/android/README.md -------------------------------------------------------------------------------- /demo/ncnn/android/app/build.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/ncnn/android/app/build.gradle -------------------------------------------------------------------------------- /demo/ncnn/android/app/src/main/AndroidManifest.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/ncnn/android/app/src/main/AndroidManifest.xml -------------------------------------------------------------------------------- /demo/ncnn/android/app/src/main/assets/yolox.param: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/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/mcmingchang/yolox_keypoint_segment/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/mcmingchang/yolox_keypoint_segment/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/mcmingchang/yolox_keypoint_segment/HEAD/demo/ncnn/android/app/src/main/jni/CMakeLists.txt -------------------------------------------------------------------------------- /demo/ncnn/android/app/src/main/jni/yoloXncnn_jni.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/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/mcmingchang/yolox_keypoint_segment/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/mcmingchang/yolox_keypoint_segment/HEAD/demo/ncnn/android/app/src/main/res/values/strings.xml -------------------------------------------------------------------------------- /demo/ncnn/android/build.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/ncnn/android/build.gradle -------------------------------------------------------------------------------- /demo/ncnn/android/gradle/wrapper/gradle-wrapper.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/ncnn/android/gradle/wrapper/gradle-wrapper.jar -------------------------------------------------------------------------------- /demo/ncnn/android/gradle/wrapper/gradle-wrapper.properties: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/ncnn/android/gradle/wrapper/gradle-wrapper.properties -------------------------------------------------------------------------------- /demo/ncnn/android/gradlew: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/ncnn/android/gradlew -------------------------------------------------------------------------------- /demo/ncnn/android/gradlew.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/ncnn/android/gradlew.bat -------------------------------------------------------------------------------- /demo/ncnn/android/settings.gradle: -------------------------------------------------------------------------------- 1 | include ':app' 2 | -------------------------------------------------------------------------------- /demo/ncnn/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/ncnn/cpp/README.md -------------------------------------------------------------------------------- /demo/ncnn/cpp/yolox.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/demo/ncnn/cpp/yolox.cpp -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- 1 | _build -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_static/css/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/docs/_static/css/custom.css -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/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/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/manipulate_training_image_size.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/docs/manipulate_training_image_size.md -------------------------------------------------------------------------------- /docs/model_zoo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/docs/model_zoo.md -------------------------------------------------------------------------------- /docs/quick_run.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/docs/quick_run.md -------------------------------------------------------------------------------- /docs/requirements-doc.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/docs/requirements-doc.txt -------------------------------------------------------------------------------- /docs/train_custom_data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/docs/train_custom_data.md -------------------------------------------------------------------------------- /docs/updates_note.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/docs/updates_note.md -------------------------------------------------------------------------------- /exps/default/nano.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/nano.py -------------------------------------------------------------------------------- /exps/default/yolov3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/yolov3.py -------------------------------------------------------------------------------- /exps/default/yolox_ccpd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/yolox_ccpd.py -------------------------------------------------------------------------------- /exps/default/yolox_decorate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/yolox_decorate.py -------------------------------------------------------------------------------- /exps/default/yolox_kp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/yolox_kp.py -------------------------------------------------------------------------------- /exps/default/yolox_l.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/yolox_l.py -------------------------------------------------------------------------------- /exps/default/yolox_m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/yolox_m.py -------------------------------------------------------------------------------- /exps/default/yolox_masonry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/yolox_masonry.py -------------------------------------------------------------------------------- /exps/default/yolox_plaster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/yolox_plaster.py -------------------------------------------------------------------------------- /exps/default/yolox_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/yolox_s.py -------------------------------------------------------------------------------- /exps/default/yolox_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/yolox_seg.py -------------------------------------------------------------------------------- /exps/default/yolox_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/yolox_tiny.py -------------------------------------------------------------------------------- /exps/default/yolox_x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/default/yolox_x.py -------------------------------------------------------------------------------- /exps/example/custom/nano.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/example/custom/nano.py -------------------------------------------------------------------------------- /exps/example/custom/yolox_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/example/custom/yolox_s.py -------------------------------------------------------------------------------- /exps/example/yolox_voc/yolox_voc_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/exps/example/yolox_voc/yolox_voc_s.py -------------------------------------------------------------------------------- /imgs/kps/5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/imgs/kps/5.jpg -------------------------------------------------------------------------------- /imgs/kps/6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/imgs/kps/6.jpg -------------------------------------------------------------------------------- /imgs/kps/7.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/imgs/kps/7.jpg -------------------------------------------------------------------------------- /imgs/segs/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/imgs/segs/1.jpg -------------------------------------------------------------------------------- /imgs/segs/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/imgs/segs/2.jpg -------------------------------------------------------------------------------- /imgs/segs/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/imgs/segs/3.jpg -------------------------------------------------------------------------------- /imgs/segs/4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/imgs/segs/4.jpg -------------------------------------------------------------------------------- /imgs/segs/5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/imgs/segs/5.jpg -------------------------------------------------------------------------------- /my_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/my_trt.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/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/mcmingchang/yolox_keypoint_segment/HEAD/tests/utils/test_model_utils.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/tools/__init__.py -------------------------------------------------------------------------------- /tools/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/tools/demo.py -------------------------------------------------------------------------------- /tools/demo_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/tools/demo_panoptic.py -------------------------------------------------------------------------------- /tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/tools/eval.py -------------------------------------------------------------------------------- /tools/export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/tools/export_onnx.py -------------------------------------------------------------------------------- /tools/export_torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/tools/export_torchscript.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/tools/trt.py -------------------------------------------------------------------------------- /trt_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/trt_convert.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/mcmingchang/yolox_keypoint_segment/HEAD/yolox/core/__init__.py -------------------------------------------------------------------------------- /yolox/core/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/core/launch.py -------------------------------------------------------------------------------- /yolox/core/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/core/trainer.py -------------------------------------------------------------------------------- /yolox/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/__init__.py -------------------------------------------------------------------------------- /yolox/data/data_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/data_augment.py -------------------------------------------------------------------------------- /yolox/data/data_prefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/data_prefetcher.py -------------------------------------------------------------------------------- /yolox/data/dataloading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/dataloading.py -------------------------------------------------------------------------------- /yolox/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/datasets/__init__.py -------------------------------------------------------------------------------- /yolox/data/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/datasets/coco.py -------------------------------------------------------------------------------- /yolox/data/datasets/coco_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/datasets/coco_classes.py -------------------------------------------------------------------------------- /yolox/data/datasets/datasets_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/datasets/datasets_wrapper.py -------------------------------------------------------------------------------- /yolox/data/datasets/generate_wall_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/datasets/generate_wall_data.py -------------------------------------------------------------------------------- /yolox/data/datasets/mosaicdetection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/datasets/mosaicdetection.py -------------------------------------------------------------------------------- /yolox/data/datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/datasets/voc.py -------------------------------------------------------------------------------- /yolox/data/datasets/voc_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/datasets/voc_classes.py -------------------------------------------------------------------------------- /yolox/data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/data/samplers.py -------------------------------------------------------------------------------- /yolox/evaluators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/evaluators/__init__.py -------------------------------------------------------------------------------- /yolox/evaluators/coco_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/evaluators/coco_evaluator.py -------------------------------------------------------------------------------- /yolox/evaluators/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/evaluators/voc_eval.py -------------------------------------------------------------------------------- /yolox/evaluators/voc_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/evaluators/voc_evaluator.py -------------------------------------------------------------------------------- /yolox/exp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/exp/__init__.py -------------------------------------------------------------------------------- /yolox/exp/base_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/exp/base_exp.py -------------------------------------------------------------------------------- /yolox/exp/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/exp/build.py -------------------------------------------------------------------------------- /yolox/exp/yolox_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/exp/yolox_base.py -------------------------------------------------------------------------------- /yolox/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/layers/__init__.py -------------------------------------------------------------------------------- /yolox/layers/fast_coco_eval_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/layers/fast_coco_eval_api.py -------------------------------------------------------------------------------- /yolox/layers/jit_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/layers/jit_ops.py -------------------------------------------------------------------------------- /yolox/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/models/__init__.py -------------------------------------------------------------------------------- /yolox/models/coatnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/models/coatnet.py -------------------------------------------------------------------------------- /yolox/models/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/models/darknet.py -------------------------------------------------------------------------------- /yolox/models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/models/losses.py -------------------------------------------------------------------------------- /yolox/models/network_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/models/network_blocks.py -------------------------------------------------------------------------------- /yolox/models/yolo_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/models/yolo_fpn.py -------------------------------------------------------------------------------- /yolox/models/yolo_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/models/yolo_head.py -------------------------------------------------------------------------------- /yolox/models/yolo_head_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/models/yolo_head_panoptic.py -------------------------------------------------------------------------------- /yolox/models/yolo_pafpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/models/yolo_pafpn.py -------------------------------------------------------------------------------- /yolox/models/yolo_pafpn_slim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/models/yolo_pafpn_slim.py -------------------------------------------------------------------------------- /yolox/models/yolov7_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/models/yolov7_tiny.py -------------------------------------------------------------------------------- /yolox/models/yolox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/models/yolox.py -------------------------------------------------------------------------------- /yolox/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/__init__.py -------------------------------------------------------------------------------- /yolox/utils/allreduce_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/allreduce_norm.py -------------------------------------------------------------------------------- /yolox/utils/boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/boxes.py -------------------------------------------------------------------------------- /yolox/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/checkpoint.py -------------------------------------------------------------------------------- /yolox/utils/compat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/compat.py -------------------------------------------------------------------------------- /yolox/utils/demo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/demo_utils.py -------------------------------------------------------------------------------- /yolox/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/dist.py -------------------------------------------------------------------------------- /yolox/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/ema.py -------------------------------------------------------------------------------- /yolox/utils/fuse_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/fuse_model.py -------------------------------------------------------------------------------- /yolox/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/logger.py -------------------------------------------------------------------------------- /yolox/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/lr_scheduler.py -------------------------------------------------------------------------------- /yolox/utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/metric.py -------------------------------------------------------------------------------- /yolox/utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/model_utils.py -------------------------------------------------------------------------------- /yolox/utils/setup_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/setup_env.py -------------------------------------------------------------------------------- /yolox/utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/yolox/utils/visualize.py -------------------------------------------------------------------------------- /注意.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mcmingchang/yolox_keypoint_segment/HEAD/注意.txt --------------------------------------------------------------------------------