├── .gitignore ├── C++_inference_openvino_kpt ├── CMakeLists.txt ├── README.md ├── demo_kpt.cpp ├── yolov7_kpt.cpp └── yolov7_kpt.h ├── DH_camera_read ├── CMakeLists.txt ├── include │ ├── CamWrapper.h │ ├── CamWrapperDH.h │ ├── DH_CAM.h │ ├── DxImageProc.h │ └── GxIAPI.h └── src │ ├── CamWrapper.cpp │ └── DH_CAM.cpp ├── LICENSE ├── README.md ├── cfg ├── armor │ ├── shufflenet-0.5-armor.yaml │ ├── shufflenet-SImAM-0.5-armor.yaml │ └── yolov8-0.5-SimAM-armor.yaml ├── board │ └── yolov8-0.5-board.yaml └── winmill │ ├── shufflenet-0.5-win.yaml │ └── yolov8-0.5-SimAM-win.yaml ├── data ├── armor │ ├── armor_detect.yaml │ ├── armor_kpt.yaml │ └── hyp.armor.yaml ├── board │ ├── board_detect.yaml │ └── hyp.board.yaml └── winmill │ ├── hyp.win.yaml │ └── win_kpt.yaml ├── detect.py ├── export.py ├── models ├── Models │ ├── mobileone.py │ └── shufflenet.py ├── __init__.py ├── basic.py ├── common.py ├── experimental.py └── yolo.py ├── pre-processing_script ├── __init__.py ├── change_anchor.py ├── change_label.py ├── change_name.py ├── change_yolo_bbox.py ├── change_yolo_kpt.py ├── label_show.py └── split_dataset.py ├── requirements.txt ├── show_pic ├── label_1.jpg ├── label_2.jpg ├── label_show_sample.jpg ├── result_openvino.jpg └── result_pytorch.jpg ├── test.py ├── train.py └── utils ├── __init__.py ├── activations.py ├── adaptive_bn.py ├── add_nms.py ├── anchor_generator.py ├── autoanchor.py ├── aws ├── __init__.py ├── mime.sh ├── resume.py └── userdata.sh ├── datasets.py ├── general.py ├── google_app_engine ├── Dockerfile ├── additional_requirements.txt └── app.yaml ├── google_utils.py ├── loss.py ├── metrics.py ├── plots.py ├── prune_utils.py ├── tal.py ├── torch_utils.py └── wandb_logging ├── __init__.py ├── log_dataset.py └── wandb_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/.gitignore -------------------------------------------------------------------------------- /C++_inference_openvino_kpt/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/C++_inference_openvino_kpt/CMakeLists.txt -------------------------------------------------------------------------------- /C++_inference_openvino_kpt/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/C++_inference_openvino_kpt/README.md -------------------------------------------------------------------------------- /C++_inference_openvino_kpt/demo_kpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/C++_inference_openvino_kpt/demo_kpt.cpp -------------------------------------------------------------------------------- /C++_inference_openvino_kpt/yolov7_kpt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/C++_inference_openvino_kpt/yolov7_kpt.cpp -------------------------------------------------------------------------------- /C++_inference_openvino_kpt/yolov7_kpt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/C++_inference_openvino_kpt/yolov7_kpt.h -------------------------------------------------------------------------------- /DH_camera_read/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/DH_camera_read/CMakeLists.txt -------------------------------------------------------------------------------- /DH_camera_read/include/CamWrapper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/DH_camera_read/include/CamWrapper.h -------------------------------------------------------------------------------- /DH_camera_read/include/CamWrapperDH.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/DH_camera_read/include/CamWrapperDH.h -------------------------------------------------------------------------------- /DH_camera_read/include/DH_CAM.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/DH_camera_read/include/DH_CAM.h -------------------------------------------------------------------------------- /DH_camera_read/include/DxImageProc.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/DH_camera_read/include/DxImageProc.h -------------------------------------------------------------------------------- /DH_camera_read/include/GxIAPI.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/DH_camera_read/include/GxIAPI.h -------------------------------------------------------------------------------- /DH_camera_read/src/CamWrapper.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/DH_camera_read/src/CamWrapper.cpp -------------------------------------------------------------------------------- /DH_camera_read/src/DH_CAM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/DH_camera_read/src/DH_CAM.cpp -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/README.md -------------------------------------------------------------------------------- /cfg/armor/shufflenet-0.5-armor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/cfg/armor/shufflenet-0.5-armor.yaml -------------------------------------------------------------------------------- /cfg/armor/shufflenet-SImAM-0.5-armor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/cfg/armor/shufflenet-SImAM-0.5-armor.yaml -------------------------------------------------------------------------------- /cfg/armor/yolov8-0.5-SimAM-armor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/cfg/armor/yolov8-0.5-SimAM-armor.yaml -------------------------------------------------------------------------------- /cfg/board/yolov8-0.5-board.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/cfg/board/yolov8-0.5-board.yaml -------------------------------------------------------------------------------- /cfg/winmill/shufflenet-0.5-win.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/cfg/winmill/shufflenet-0.5-win.yaml -------------------------------------------------------------------------------- /cfg/winmill/yolov8-0.5-SimAM-win.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/cfg/winmill/yolov8-0.5-SimAM-win.yaml -------------------------------------------------------------------------------- /data/armor/armor_detect.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/data/armor/armor_detect.yaml -------------------------------------------------------------------------------- /data/armor/armor_kpt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/data/armor/armor_kpt.yaml -------------------------------------------------------------------------------- /data/armor/hyp.armor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/data/armor/hyp.armor.yaml -------------------------------------------------------------------------------- /data/board/board_detect.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/data/board/board_detect.yaml -------------------------------------------------------------------------------- /data/board/hyp.board.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/data/board/hyp.board.yaml -------------------------------------------------------------------------------- /data/winmill/hyp.win.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/data/winmill/hyp.win.yaml -------------------------------------------------------------------------------- /data/winmill/win_kpt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/data/winmill/win_kpt.yaml -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/detect.py -------------------------------------------------------------------------------- /export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/export.py -------------------------------------------------------------------------------- /models/Models/mobileone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/models/Models/mobileone.py -------------------------------------------------------------------------------- /models/Models/shufflenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/models/Models/shufflenet.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | # inits -------------------------------------------------------------------------------- /models/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/models/basic.py -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/models/common.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/models/yolo.py -------------------------------------------------------------------------------- /pre-processing_script/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pre-processing_script/change_anchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/pre-processing_script/change_anchor.py -------------------------------------------------------------------------------- /pre-processing_script/change_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/pre-processing_script/change_label.py -------------------------------------------------------------------------------- /pre-processing_script/change_name.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/pre-processing_script/change_name.py -------------------------------------------------------------------------------- /pre-processing_script/change_yolo_bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/pre-processing_script/change_yolo_bbox.py -------------------------------------------------------------------------------- /pre-processing_script/change_yolo_kpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/pre-processing_script/change_yolo_kpt.py -------------------------------------------------------------------------------- /pre-processing_script/label_show.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/pre-processing_script/label_show.py -------------------------------------------------------------------------------- /pre-processing_script/split_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/pre-processing_script/split_dataset.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/requirements.txt -------------------------------------------------------------------------------- /show_pic/label_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/show_pic/label_1.jpg -------------------------------------------------------------------------------- /show_pic/label_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/show_pic/label_2.jpg -------------------------------------------------------------------------------- /show_pic/label_show_sample.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/show_pic/label_show_sample.jpg -------------------------------------------------------------------------------- /show_pic/result_openvino.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/show_pic/result_openvino.jpg -------------------------------------------------------------------------------- /show_pic/result_pytorch.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/show_pic/result_pytorch.jpg -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | # init 2 | -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/adaptive_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/adaptive_bn.py -------------------------------------------------------------------------------- /utils/add_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/add_nms.py -------------------------------------------------------------------------------- /utils/anchor_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/anchor_generator.py -------------------------------------------------------------------------------- /utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/autoanchor.py -------------------------------------------------------------------------------- /utils/aws/__init__.py: -------------------------------------------------------------------------------- 1 | # init -------------------------------------------------------------------------------- /utils/aws/mime.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/aws/mime.sh -------------------------------------------------------------------------------- /utils/aws/resume.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/aws/resume.py -------------------------------------------------------------------------------- /utils/aws/userdata.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/aws/userdata.sh -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_app_engine/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/google_app_engine/Dockerfile -------------------------------------------------------------------------------- /utils/google_app_engine/additional_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/google_app_engine/additional_requirements.txt -------------------------------------------------------------------------------- /utils/google_app_engine/app.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/google_app_engine/app.yaml -------------------------------------------------------------------------------- /utils/google_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/google_utils.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/prune_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/prune_utils.py -------------------------------------------------------------------------------- /utils/tal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/tal.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /utils/wandb_logging/__init__.py: -------------------------------------------------------------------------------- 1 | # init -------------------------------------------------------------------------------- /utils/wandb_logging/log_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/wandb_logging/log_dataset.py -------------------------------------------------------------------------------- /utils/wandb_logging/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023/HEAD/utils/wandb_logging/wandb_utils.py --------------------------------------------------------------------------------