├── .github └── ISSUE_TEMPLATE │ ├── config.yml │ ├── feature_request.yml │ └── question.yml ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── README_cn.md ├── assets ├── banner-YOLO.png ├── image3.jpg ├── picture.png ├── speed_comparision_v2.png ├── speed_comparision_v3.png ├── train_batch.jpg ├── voc_loss_curve.jpg ├── wechat_qrcode.png ├── yolov5s.jpg ├── yolov6s.jpg └── yoloxs.jpg ├── configs ├── base │ ├── README.md │ ├── README_cn.md │ ├── yolov6l_base.py │ ├── yolov6l_base_finetune.py │ ├── yolov6m_base.py │ ├── yolov6m_base_finetune.py │ ├── yolov6n_base.py │ ├── yolov6n_base_finetune.py │ ├── yolov6s_base.py │ └── yolov6s_base_finetune.py ├── experiment │ ├── eval_640_repro.py │ ├── yolov6n_with_eval_params.py │ ├── yolov6s_csp_scaled.py │ ├── yolov6t.py │ ├── yolov6t_csp_scaled.py │ └── yolov6t_finetune.py ├── qarepvgg │ ├── README.md │ ├── yolov6m_qa.py │ ├── yolov6n_qa.py │ └── yolov6s_qa.py ├── repopt │ ├── yolov6_tiny_hs.py │ ├── yolov6_tiny_opt.py │ ├── yolov6_tiny_opt_qat.py │ ├── yolov6n_hs.py │ ├── yolov6n_opt.py │ ├── yolov6n_opt_qat.py │ ├── yolov6s_hs.py │ ├── yolov6s_opt.py │ └── yolov6s_opt_qat.py ├── yolov6l.py ├── yolov6l6.py ├── yolov6l6_finetune.py ├── yolov6l_finetune.py ├── yolov6m.py ├── yolov6m6.py ├── yolov6m6_finetune.py ├── yolov6m_finetune.py ├── yolov6n.py ├── yolov6n6.py ├── yolov6n6_finetune.py ├── yolov6n_finetune.py ├── yolov6s.py ├── yolov6s6.py ├── yolov6s6_finetune.py └── yolov6s_finetune.py ├── data ├── coco.yaml ├── dataset.yaml ├── images │ ├── image1.jpg │ ├── image2.jpg │ └── image3.jpg └── voc.yaml ├── deploy ├── ONNX │ ├── OpenCV │ │ ├── README.md │ │ ├── coco.names │ │ ├── sample.jpg │ │ ├── yolo.py │ │ ├── yolo_video.py │ │ ├── yolov5 │ │ │ ├── CMakeLists.txt │ │ │ └── yolov5.cpp │ │ ├── yolov6 │ │ │ ├── CMakeLists.txt │ │ │ └── yolov6.cpp │ │ ├── yolox.py │ │ └── yolox │ │ │ ├── CMakeLists.txt │ │ │ └── yolox.cpp │ ├── README.md │ ├── YOLOv6-Dynamic-Batch-onnxruntime.ipynb │ ├── YOLOv6-Dynamic-Batch-tensorrt.ipynb │ ├── eval_trt.py │ └── export_onnx.py ├── OpenVINO │ ├── README.md │ └── export_openvino.py ├── RKNN │ ├── RKOPT_README.md │ ├── RKOPT_README_cn.md │ └── export_onnx_for_rknn.py └── TensorRT │ ├── CMakeLists.txt │ ├── README.md │ ├── calibrator.py │ ├── eval_yolo_trt.py │ ├── logging.h │ ├── onnx_to_trt.py │ ├── tensorrt_processor.py │ ├── visualize.py │ └── yolov6.cpp ├── docs ├── Test_speed.md ├── Train_coco_data.md ├── Train_custom_data.md ├── Tutorial of Quantization.md ├── tutorial_repopt.md └── tutorial_voc.ipynb ├── hubconf.py ├── inference.ipynb ├── requirements.txt ├── tools ├── eval.py ├── infer.py ├── partial_quantization │ ├── README.md │ ├── eval.py │ ├── eval.yaml │ ├── partial_quant.py │ ├── ptq.py │ ├── sensitivity_analyse.py │ └── utils.py ├── qat │ ├── README.md │ ├── onnx_utils.py │ ├── qat_export.py │ └── qat_utils.py ├── quantization │ ├── mnn │ │ └── README.md │ ├── ppq │ │ ├── ProgramEntrance.py │ │ └── write_qparams_onnx2trt.py │ └── tensorrt │ │ ├── post_training │ │ ├── Calibrator.py │ │ ├── LICENSE │ │ ├── README.md │ │ ├── onnx_to_tensorrt.py │ │ └── quant.sh │ │ ├── requirements.txt │ │ └── training_aware │ │ └── QAT_quantizer.py └── train.py ├── turtorial.ipynb └── yolov6 ├── __init__.py ├── assigners ├── __init__.py ├── anchor_generator.py ├── assigner_utils.py ├── atss_assigner.py ├── iou2d_calculator.py └── tal_assigner.py ├── core ├── engine.py ├── evaler.py └── inferer.py ├── data ├── data_augment.py ├── data_load.py ├── datasets.py ├── vis_dataset.py └── voc2yolo.py ├── layers ├── common.py └── dbb_transforms.py ├── models ├── efficientrep.py ├── effidehead.py ├── end2end.py ├── heads │ ├── effidehead_distill_ns.py │ └── effidehead_fuseab.py ├── losses │ ├── loss.py │ ├── loss_distill.py │ ├── loss_distill_ns.py │ └── loss_fuseab.py ├── reppan.py └── yolo.py ├── solver └── build.py └── utils ├── Arial.ttf ├── RepOptimizer.py ├── checkpoint.py ├── config.py ├── ema.py ├── envs.py ├── events.py ├── figure_iou.py ├── general.py ├── metrics.py ├── nms.py └── torch_utils.py /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- 1 | blank_issues_enabled: false 2 | -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/.github/ISSUE_TEMPLATE/feature_request.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/question.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/.github/ISSUE_TEMPLATE/question.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/README.md -------------------------------------------------------------------------------- /README_cn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/README_cn.md -------------------------------------------------------------------------------- /assets/banner-YOLO.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/assets/banner-YOLO.png -------------------------------------------------------------------------------- /assets/image3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/assets/image3.jpg -------------------------------------------------------------------------------- /assets/picture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/assets/picture.png -------------------------------------------------------------------------------- /assets/speed_comparision_v2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/assets/speed_comparision_v2.png -------------------------------------------------------------------------------- /assets/speed_comparision_v3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/assets/speed_comparision_v3.png -------------------------------------------------------------------------------- /assets/train_batch.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/assets/train_batch.jpg -------------------------------------------------------------------------------- /assets/voc_loss_curve.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/assets/voc_loss_curve.jpg -------------------------------------------------------------------------------- /assets/wechat_qrcode.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/assets/wechat_qrcode.png -------------------------------------------------------------------------------- /assets/yolov5s.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/assets/yolov5s.jpg -------------------------------------------------------------------------------- /assets/yolov6s.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/assets/yolov6s.jpg -------------------------------------------------------------------------------- /assets/yoloxs.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/assets/yoloxs.jpg -------------------------------------------------------------------------------- /configs/base/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/base/README.md -------------------------------------------------------------------------------- /configs/base/README_cn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/base/README_cn.md -------------------------------------------------------------------------------- /configs/base/yolov6l_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/base/yolov6l_base.py -------------------------------------------------------------------------------- /configs/base/yolov6l_base_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/base/yolov6l_base_finetune.py -------------------------------------------------------------------------------- /configs/base/yolov6m_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/base/yolov6m_base.py -------------------------------------------------------------------------------- /configs/base/yolov6m_base_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/base/yolov6m_base_finetune.py -------------------------------------------------------------------------------- /configs/base/yolov6n_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/base/yolov6n_base.py -------------------------------------------------------------------------------- /configs/base/yolov6n_base_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/base/yolov6n_base_finetune.py -------------------------------------------------------------------------------- /configs/base/yolov6s_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/base/yolov6s_base.py -------------------------------------------------------------------------------- /configs/base/yolov6s_base_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/base/yolov6s_base_finetune.py -------------------------------------------------------------------------------- /configs/experiment/eval_640_repro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/experiment/eval_640_repro.py -------------------------------------------------------------------------------- /configs/experiment/yolov6n_with_eval_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/experiment/yolov6n_with_eval_params.py -------------------------------------------------------------------------------- /configs/experiment/yolov6s_csp_scaled.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/experiment/yolov6s_csp_scaled.py -------------------------------------------------------------------------------- /configs/experiment/yolov6t.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/experiment/yolov6t.py -------------------------------------------------------------------------------- /configs/experiment/yolov6t_csp_scaled.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/experiment/yolov6t_csp_scaled.py -------------------------------------------------------------------------------- /configs/experiment/yolov6t_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/experiment/yolov6t_finetune.py -------------------------------------------------------------------------------- /configs/qarepvgg/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/qarepvgg/README.md -------------------------------------------------------------------------------- /configs/qarepvgg/yolov6m_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/qarepvgg/yolov6m_qa.py -------------------------------------------------------------------------------- /configs/qarepvgg/yolov6n_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/qarepvgg/yolov6n_qa.py -------------------------------------------------------------------------------- /configs/qarepvgg/yolov6s_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/qarepvgg/yolov6s_qa.py -------------------------------------------------------------------------------- /configs/repopt/yolov6_tiny_hs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/repopt/yolov6_tiny_hs.py -------------------------------------------------------------------------------- /configs/repopt/yolov6_tiny_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/repopt/yolov6_tiny_opt.py -------------------------------------------------------------------------------- /configs/repopt/yolov6_tiny_opt_qat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/repopt/yolov6_tiny_opt_qat.py -------------------------------------------------------------------------------- /configs/repopt/yolov6n_hs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/repopt/yolov6n_hs.py -------------------------------------------------------------------------------- /configs/repopt/yolov6n_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/repopt/yolov6n_opt.py -------------------------------------------------------------------------------- /configs/repopt/yolov6n_opt_qat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/repopt/yolov6n_opt_qat.py -------------------------------------------------------------------------------- /configs/repopt/yolov6s_hs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/repopt/yolov6s_hs.py -------------------------------------------------------------------------------- /configs/repopt/yolov6s_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/repopt/yolov6s_opt.py -------------------------------------------------------------------------------- /configs/repopt/yolov6s_opt_qat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/repopt/yolov6s_opt_qat.py -------------------------------------------------------------------------------- /configs/yolov6l.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6l.py -------------------------------------------------------------------------------- /configs/yolov6l6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6l6.py -------------------------------------------------------------------------------- /configs/yolov6l6_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6l6_finetune.py -------------------------------------------------------------------------------- /configs/yolov6l_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6l_finetune.py -------------------------------------------------------------------------------- /configs/yolov6m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6m.py -------------------------------------------------------------------------------- /configs/yolov6m6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6m6.py -------------------------------------------------------------------------------- /configs/yolov6m6_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6m6_finetune.py -------------------------------------------------------------------------------- /configs/yolov6m_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6m_finetune.py -------------------------------------------------------------------------------- /configs/yolov6n.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6n.py -------------------------------------------------------------------------------- /configs/yolov6n6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6n6.py -------------------------------------------------------------------------------- /configs/yolov6n6_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6n6_finetune.py -------------------------------------------------------------------------------- /configs/yolov6n_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6n_finetune.py -------------------------------------------------------------------------------- /configs/yolov6s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6s.py -------------------------------------------------------------------------------- /configs/yolov6s6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6s6.py -------------------------------------------------------------------------------- /configs/yolov6s6_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6s6_finetune.py -------------------------------------------------------------------------------- /configs/yolov6s_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/configs/yolov6s_finetune.py -------------------------------------------------------------------------------- /data/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/data/coco.yaml -------------------------------------------------------------------------------- /data/dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/data/dataset.yaml -------------------------------------------------------------------------------- /data/images/image1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/data/images/image1.jpg -------------------------------------------------------------------------------- /data/images/image2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/data/images/image2.jpg -------------------------------------------------------------------------------- /data/images/image3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/data/images/image3.jpg -------------------------------------------------------------------------------- /data/voc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/data/voc.yaml -------------------------------------------------------------------------------- /deploy/ONNX/OpenCV/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/OpenCV/README.md -------------------------------------------------------------------------------- /deploy/ONNX/OpenCV/coco.names: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/OpenCV/coco.names -------------------------------------------------------------------------------- /deploy/ONNX/OpenCV/sample.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/OpenCV/sample.jpg -------------------------------------------------------------------------------- /deploy/ONNX/OpenCV/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/OpenCV/yolo.py -------------------------------------------------------------------------------- /deploy/ONNX/OpenCV/yolo_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/OpenCV/yolo_video.py -------------------------------------------------------------------------------- /deploy/ONNX/OpenCV/yolov5/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/OpenCV/yolov5/CMakeLists.txt -------------------------------------------------------------------------------- /deploy/ONNX/OpenCV/yolov5/yolov5.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/OpenCV/yolov5/yolov5.cpp -------------------------------------------------------------------------------- /deploy/ONNX/OpenCV/yolov6/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/OpenCV/yolov6/CMakeLists.txt -------------------------------------------------------------------------------- /deploy/ONNX/OpenCV/yolov6/yolov6.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/OpenCV/yolov6/yolov6.cpp -------------------------------------------------------------------------------- /deploy/ONNX/OpenCV/yolox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/OpenCV/yolox.py -------------------------------------------------------------------------------- /deploy/ONNX/OpenCV/yolox/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/OpenCV/yolox/CMakeLists.txt -------------------------------------------------------------------------------- /deploy/ONNX/OpenCV/yolox/yolox.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/OpenCV/yolox/yolox.cpp -------------------------------------------------------------------------------- /deploy/ONNX/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/README.md -------------------------------------------------------------------------------- /deploy/ONNX/YOLOv6-Dynamic-Batch-onnxruntime.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/YOLOv6-Dynamic-Batch-onnxruntime.ipynb -------------------------------------------------------------------------------- /deploy/ONNX/YOLOv6-Dynamic-Batch-tensorrt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/YOLOv6-Dynamic-Batch-tensorrt.ipynb -------------------------------------------------------------------------------- /deploy/ONNX/eval_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/eval_trt.py -------------------------------------------------------------------------------- /deploy/ONNX/export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/ONNX/export_onnx.py -------------------------------------------------------------------------------- /deploy/OpenVINO/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/OpenVINO/README.md -------------------------------------------------------------------------------- /deploy/OpenVINO/export_openvino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/OpenVINO/export_openvino.py -------------------------------------------------------------------------------- /deploy/RKNN/RKOPT_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/RKNN/RKOPT_README.md -------------------------------------------------------------------------------- /deploy/RKNN/RKOPT_README_cn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/RKNN/RKOPT_README_cn.md -------------------------------------------------------------------------------- /deploy/RKNN/export_onnx_for_rknn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/RKNN/export_onnx_for_rknn.py -------------------------------------------------------------------------------- /deploy/TensorRT/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/TensorRT/CMakeLists.txt -------------------------------------------------------------------------------- /deploy/TensorRT/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/TensorRT/README.md -------------------------------------------------------------------------------- /deploy/TensorRT/calibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/TensorRT/calibrator.py -------------------------------------------------------------------------------- /deploy/TensorRT/eval_yolo_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/TensorRT/eval_yolo_trt.py -------------------------------------------------------------------------------- /deploy/TensorRT/logging.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/TensorRT/logging.h -------------------------------------------------------------------------------- /deploy/TensorRT/onnx_to_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/TensorRT/onnx_to_trt.py -------------------------------------------------------------------------------- /deploy/TensorRT/tensorrt_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/TensorRT/tensorrt_processor.py -------------------------------------------------------------------------------- /deploy/TensorRT/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/TensorRT/visualize.py -------------------------------------------------------------------------------- /deploy/TensorRT/yolov6.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/deploy/TensorRT/yolov6.cpp -------------------------------------------------------------------------------- /docs/Test_speed.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/docs/Test_speed.md -------------------------------------------------------------------------------- /docs/Train_coco_data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/docs/Train_coco_data.md -------------------------------------------------------------------------------- /docs/Train_custom_data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/docs/Train_custom_data.md -------------------------------------------------------------------------------- /docs/Tutorial of Quantization.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/docs/Tutorial of Quantization.md -------------------------------------------------------------------------------- /docs/tutorial_repopt.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/docs/tutorial_repopt.md -------------------------------------------------------------------------------- /docs/tutorial_voc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/docs/tutorial_voc.ipynb -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/hubconf.py -------------------------------------------------------------------------------- /inference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/inference.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/eval.py -------------------------------------------------------------------------------- /tools/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/infer.py -------------------------------------------------------------------------------- /tools/partial_quantization/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/partial_quantization/README.md -------------------------------------------------------------------------------- /tools/partial_quantization/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/partial_quantization/eval.py -------------------------------------------------------------------------------- /tools/partial_quantization/eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/partial_quantization/eval.yaml -------------------------------------------------------------------------------- /tools/partial_quantization/partial_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/partial_quantization/partial_quant.py -------------------------------------------------------------------------------- /tools/partial_quantization/ptq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/partial_quantization/ptq.py -------------------------------------------------------------------------------- /tools/partial_quantization/sensitivity_analyse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/partial_quantization/sensitivity_analyse.py -------------------------------------------------------------------------------- /tools/partial_quantization/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/partial_quantization/utils.py -------------------------------------------------------------------------------- /tools/qat/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/qat/README.md -------------------------------------------------------------------------------- /tools/qat/onnx_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/qat/onnx_utils.py -------------------------------------------------------------------------------- /tools/qat/qat_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/qat/qat_export.py -------------------------------------------------------------------------------- /tools/qat/qat_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/qat/qat_utils.py -------------------------------------------------------------------------------- /tools/quantization/mnn/README.md: -------------------------------------------------------------------------------- 1 | # Coming soon 2 | -------------------------------------------------------------------------------- /tools/quantization/ppq/ProgramEntrance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/quantization/ppq/ProgramEntrance.py -------------------------------------------------------------------------------- /tools/quantization/ppq/write_qparams_onnx2trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/quantization/ppq/write_qparams_onnx2trt.py -------------------------------------------------------------------------------- /tools/quantization/tensorrt/post_training/Calibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/quantization/tensorrt/post_training/Calibrator.py -------------------------------------------------------------------------------- /tools/quantization/tensorrt/post_training/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/quantization/tensorrt/post_training/LICENSE -------------------------------------------------------------------------------- /tools/quantization/tensorrt/post_training/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/quantization/tensorrt/post_training/README.md -------------------------------------------------------------------------------- /tools/quantization/tensorrt/post_training/onnx_to_tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/quantization/tensorrt/post_training/onnx_to_tensorrt.py -------------------------------------------------------------------------------- /tools/quantization/tensorrt/post_training/quant.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/quantization/tensorrt/post_training/quant.sh -------------------------------------------------------------------------------- /tools/quantization/tensorrt/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/quantization/tensorrt/requirements.txt -------------------------------------------------------------------------------- /tools/quantization/tensorrt/training_aware/QAT_quantizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/quantization/tensorrt/training_aware/QAT_quantizer.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/tools/train.py -------------------------------------------------------------------------------- /turtorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/turtorial.ipynb -------------------------------------------------------------------------------- /yolov6/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /yolov6/assigners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/assigners/__init__.py -------------------------------------------------------------------------------- /yolov6/assigners/anchor_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/assigners/anchor_generator.py -------------------------------------------------------------------------------- /yolov6/assigners/assigner_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/assigners/assigner_utils.py -------------------------------------------------------------------------------- /yolov6/assigners/atss_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/assigners/atss_assigner.py -------------------------------------------------------------------------------- /yolov6/assigners/iou2d_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/assigners/iou2d_calculator.py -------------------------------------------------------------------------------- /yolov6/assigners/tal_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/assigners/tal_assigner.py -------------------------------------------------------------------------------- /yolov6/core/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/core/engine.py -------------------------------------------------------------------------------- /yolov6/core/evaler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/core/evaler.py -------------------------------------------------------------------------------- /yolov6/core/inferer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/core/inferer.py -------------------------------------------------------------------------------- /yolov6/data/data_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/data/data_augment.py -------------------------------------------------------------------------------- /yolov6/data/data_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/data/data_load.py -------------------------------------------------------------------------------- /yolov6/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/data/datasets.py -------------------------------------------------------------------------------- /yolov6/data/vis_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/data/vis_dataset.py -------------------------------------------------------------------------------- /yolov6/data/voc2yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/data/voc2yolo.py -------------------------------------------------------------------------------- /yolov6/layers/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/layers/common.py -------------------------------------------------------------------------------- /yolov6/layers/dbb_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/layers/dbb_transforms.py -------------------------------------------------------------------------------- /yolov6/models/efficientrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/models/efficientrep.py -------------------------------------------------------------------------------- /yolov6/models/effidehead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/models/effidehead.py -------------------------------------------------------------------------------- /yolov6/models/end2end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/models/end2end.py -------------------------------------------------------------------------------- /yolov6/models/heads/effidehead_distill_ns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/models/heads/effidehead_distill_ns.py -------------------------------------------------------------------------------- /yolov6/models/heads/effidehead_fuseab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/models/heads/effidehead_fuseab.py -------------------------------------------------------------------------------- /yolov6/models/losses/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/models/losses/loss.py -------------------------------------------------------------------------------- /yolov6/models/losses/loss_distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/models/losses/loss_distill.py -------------------------------------------------------------------------------- /yolov6/models/losses/loss_distill_ns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/models/losses/loss_distill_ns.py -------------------------------------------------------------------------------- /yolov6/models/losses/loss_fuseab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/models/losses/loss_fuseab.py -------------------------------------------------------------------------------- /yolov6/models/reppan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/models/reppan.py -------------------------------------------------------------------------------- /yolov6/models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/models/yolo.py -------------------------------------------------------------------------------- /yolov6/solver/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/solver/build.py -------------------------------------------------------------------------------- /yolov6/utils/Arial.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/utils/Arial.ttf -------------------------------------------------------------------------------- /yolov6/utils/RepOptimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/utils/RepOptimizer.py -------------------------------------------------------------------------------- /yolov6/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/utils/checkpoint.py -------------------------------------------------------------------------------- /yolov6/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/utils/config.py -------------------------------------------------------------------------------- /yolov6/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/utils/ema.py -------------------------------------------------------------------------------- /yolov6/utils/envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/utils/envs.py -------------------------------------------------------------------------------- /yolov6/utils/events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/utils/events.py -------------------------------------------------------------------------------- /yolov6/utils/figure_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/utils/figure_iou.py -------------------------------------------------------------------------------- /yolov6/utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/utils/general.py -------------------------------------------------------------------------------- /yolov6/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/utils/metrics.py -------------------------------------------------------------------------------- /yolov6/utils/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/utils/nms.py -------------------------------------------------------------------------------- /yolov6/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/airockchip/YOLOv6/HEAD/yolov6/utils/torch_utils.py --------------------------------------------------------------------------------