├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.md │ └── feature_request.md └── workflows │ ├── is.yml │ └── pre-commit.yml ├── .gitignore ├── .pre-commit-config.yaml ├── Dockerfile ├── LICENSE ├── README.md ├── README_cn.md ├── README_ja.md ├── configs ├── dataset │ ├── coco_detection.yml │ ├── crowdhuman_detection.yml │ ├── custom_detection.yml │ ├── obj365_detection.yml │ └── voc_detection.yml ├── dfine │ ├── crowdhuman │ │ ├── dfine_hgnetv2_l_ch.yml │ │ ├── dfine_hgnetv2_m_ch.yml │ │ ├── dfine_hgnetv2_n_ch.yml │ │ ├── dfine_hgnetv2_s_ch.yml │ │ └── dfine_hgnetv2_x_ch.yml │ ├── custom │ │ ├── dfine_hgnetv2_l_custom.yml │ │ ├── dfine_hgnetv2_m_custom.yml │ │ ├── dfine_hgnetv2_n_custom.yml │ │ ├── dfine_hgnetv2_s_custom.yml │ │ ├── dfine_hgnetv2_x_custom.yml │ │ └── objects365 │ │ │ ├── dfine_hgnetv2_l_obj2custom.yml │ │ │ ├── dfine_hgnetv2_m_obj2custom.yml │ │ │ ├── dfine_hgnetv2_s_obj2custom.yml │ │ │ └── dfine_hgnetv2_x_obj2custom.yml │ ├── dfine_hgnetv2_l_coco.yml │ ├── dfine_hgnetv2_m_coco.yml │ ├── dfine_hgnetv2_n_coco.yml │ ├── dfine_hgnetv2_s_coco.yml │ ├── dfine_hgnetv2_x_coco.yml │ ├── include │ │ ├── dataloader.yml │ │ ├── dfine_hgnetv2.yml │ │ └── optimizer.yml │ └── objects365 │ │ ├── dfine_hgnetv2_l_obj2coco.yml │ │ ├── dfine_hgnetv2_l_obj365.yml │ │ ├── dfine_hgnetv2_m_obj2coco.yml │ │ ├── dfine_hgnetv2_m_obj365.yml │ │ ├── dfine_hgnetv2_n_obj2coco.yml │ │ ├── dfine_hgnetv2_n_obj365.yml │ │ ├── dfine_hgnetv2_s_obj2coco.yml │ │ ├── dfine_hgnetv2_s_obj365.yml │ │ ├── dfine_hgnetv2_x_obj2coco.yml │ │ └── dfine_hgnetv2_x_obj365.yml └── runtime.yml ├── reference ├── convert_weight.py └── safe_training.sh ├── requirements.txt ├── src ├── __init__.py ├── core │ ├── __init__.py │ ├── _config.py │ ├── workspace.py │ ├── yaml_config.py │ └── yaml_utils.py ├── data │ ├── __init__.py │ ├── _misc.py │ ├── dataloader.py │ ├── dataset │ │ ├── __init__.py │ │ ├── _dataset.py │ │ ├── cifar_dataset.py │ │ ├── coco_dataset.py │ │ ├── coco_eval.py │ │ ├── coco_utils.py │ │ ├── voc_detection.py │ │ └── voc_eval.py │ └── transforms │ │ ├── __init__.py │ │ ├── _transforms.py │ │ ├── container.py │ │ ├── functional.py │ │ ├── mosaic.py │ │ └── presets.py ├── misc │ ├── __init__.py │ ├── box_ops.py │ ├── dist_utils.py │ ├── lazy_loader.py │ ├── logger.py │ ├── profiler_utils.py │ └── visualizer.py ├── nn │ ├── __init__.py │ ├── arch │ │ ├── __init__.py │ │ ├── classification.py │ │ └── yolo.py │ ├── backbone │ │ ├── __init__.py │ │ ├── common.py │ │ ├── csp_darknet.py │ │ ├── csp_resnet.py │ │ ├── hgnetv2.py │ │ ├── presnet.py │ │ ├── test_resnet.py │ │ ├── timm_model.py │ │ ├── torchvision_model.py │ │ └── utils.py │ ├── criterion │ │ ├── __init__.py │ │ └── det_criterion.py │ └── postprocessor │ │ ├── __init__.py │ │ ├── box_revert.py │ │ ├── detr_postprocessor.py │ │ └── nms_postprocessor.py ├── optim │ ├── __init__.py │ ├── amp.py │ ├── ema.py │ ├── optim.py │ └── warmup.py ├── solver │ ├── __init__.py │ ├── _solver.py │ ├── clas_engine.py │ ├── clas_solver.py │ ├── det_engine.py │ ├── det_solver.py │ └── validator.py └── zoo │ ├── __init__.py │ └── dfine │ ├── __init__.py │ ├── blog.md │ ├── blog_cn.md │ ├── box_ops.py │ ├── denoising.py │ ├── dfine.py │ ├── dfine_criterion.py │ ├── dfine_decoder.py │ ├── dfine_utils.py │ ├── hybrid_encoder.py │ ├── matcher.py │ ├── postprocessor.py │ └── utils.py ├── tools ├── benchmark │ ├── dataset.py │ ├── get_info.py │ ├── requirements.txt │ ├── trt_benchmark.py │ └── utils.py ├── dataset │ ├── remap_obj365.py │ └── resize_obj365.py ├── deployment │ ├── export_onnx.py │ └── export_yolo_w_nms.py ├── inference │ ├── cppExample │ │ ├── onnx │ │ │ ├── CMakeLists.txt │ │ │ └── onnxExample.cpp │ │ ├── openvino │ │ │ ├── CMakeLists.txt │ │ │ └── openvinoExample.cpp │ │ └── trt │ │ │ ├── CMakeLists.txt │ │ │ └── trtExample.cpp │ ├── onnx_inf.py │ ├── openvino_inf.py │ ├── requirements.txt │ ├── torch_inf.py │ └── trt_inf.py └── visualization │ └── fiftyone_vis.py └── train.py /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/workflows/is.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/.github/workflows/is.yml -------------------------------------------------------------------------------- /.github/workflows/pre-commit.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/.github/workflows/pre-commit.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/README.md -------------------------------------------------------------------------------- /README_cn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/README_cn.md -------------------------------------------------------------------------------- /README_ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/README_ja.md -------------------------------------------------------------------------------- /configs/dataset/coco_detection.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dataset/coco_detection.yml -------------------------------------------------------------------------------- /configs/dataset/crowdhuman_detection.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dataset/crowdhuman_detection.yml -------------------------------------------------------------------------------- /configs/dataset/custom_detection.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dataset/custom_detection.yml -------------------------------------------------------------------------------- /configs/dataset/obj365_detection.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dataset/obj365_detection.yml -------------------------------------------------------------------------------- /configs/dataset/voc_detection.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dataset/voc_detection.yml -------------------------------------------------------------------------------- /configs/dfine/crowdhuman/dfine_hgnetv2_l_ch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/crowdhuman/dfine_hgnetv2_l_ch.yml -------------------------------------------------------------------------------- /configs/dfine/crowdhuman/dfine_hgnetv2_m_ch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/crowdhuman/dfine_hgnetv2_m_ch.yml -------------------------------------------------------------------------------- /configs/dfine/crowdhuman/dfine_hgnetv2_n_ch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/crowdhuman/dfine_hgnetv2_n_ch.yml -------------------------------------------------------------------------------- /configs/dfine/crowdhuman/dfine_hgnetv2_s_ch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/crowdhuman/dfine_hgnetv2_s_ch.yml -------------------------------------------------------------------------------- /configs/dfine/crowdhuman/dfine_hgnetv2_x_ch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/crowdhuman/dfine_hgnetv2_x_ch.yml -------------------------------------------------------------------------------- /configs/dfine/custom/dfine_hgnetv2_l_custom.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/custom/dfine_hgnetv2_l_custom.yml -------------------------------------------------------------------------------- /configs/dfine/custom/dfine_hgnetv2_m_custom.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/custom/dfine_hgnetv2_m_custom.yml -------------------------------------------------------------------------------- /configs/dfine/custom/dfine_hgnetv2_n_custom.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/custom/dfine_hgnetv2_n_custom.yml -------------------------------------------------------------------------------- /configs/dfine/custom/dfine_hgnetv2_s_custom.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/custom/dfine_hgnetv2_s_custom.yml -------------------------------------------------------------------------------- /configs/dfine/custom/dfine_hgnetv2_x_custom.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/custom/dfine_hgnetv2_x_custom.yml -------------------------------------------------------------------------------- /configs/dfine/custom/objects365/dfine_hgnetv2_l_obj2custom.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/custom/objects365/dfine_hgnetv2_l_obj2custom.yml -------------------------------------------------------------------------------- /configs/dfine/custom/objects365/dfine_hgnetv2_m_obj2custom.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/custom/objects365/dfine_hgnetv2_m_obj2custom.yml -------------------------------------------------------------------------------- /configs/dfine/custom/objects365/dfine_hgnetv2_s_obj2custom.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/custom/objects365/dfine_hgnetv2_s_obj2custom.yml -------------------------------------------------------------------------------- /configs/dfine/custom/objects365/dfine_hgnetv2_x_obj2custom.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/custom/objects365/dfine_hgnetv2_x_obj2custom.yml -------------------------------------------------------------------------------- /configs/dfine/dfine_hgnetv2_l_coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/dfine_hgnetv2_l_coco.yml -------------------------------------------------------------------------------- /configs/dfine/dfine_hgnetv2_m_coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/dfine_hgnetv2_m_coco.yml -------------------------------------------------------------------------------- /configs/dfine/dfine_hgnetv2_n_coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/dfine_hgnetv2_n_coco.yml -------------------------------------------------------------------------------- /configs/dfine/dfine_hgnetv2_s_coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/dfine_hgnetv2_s_coco.yml -------------------------------------------------------------------------------- /configs/dfine/dfine_hgnetv2_x_coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/dfine_hgnetv2_x_coco.yml -------------------------------------------------------------------------------- /configs/dfine/include/dataloader.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/include/dataloader.yml -------------------------------------------------------------------------------- /configs/dfine/include/dfine_hgnetv2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/include/dfine_hgnetv2.yml -------------------------------------------------------------------------------- /configs/dfine/include/optimizer.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/include/optimizer.yml -------------------------------------------------------------------------------- /configs/dfine/objects365/dfine_hgnetv2_l_obj2coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/objects365/dfine_hgnetv2_l_obj2coco.yml -------------------------------------------------------------------------------- /configs/dfine/objects365/dfine_hgnetv2_l_obj365.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/objects365/dfine_hgnetv2_l_obj365.yml -------------------------------------------------------------------------------- /configs/dfine/objects365/dfine_hgnetv2_m_obj2coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/objects365/dfine_hgnetv2_m_obj2coco.yml -------------------------------------------------------------------------------- /configs/dfine/objects365/dfine_hgnetv2_m_obj365.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/objects365/dfine_hgnetv2_m_obj365.yml -------------------------------------------------------------------------------- /configs/dfine/objects365/dfine_hgnetv2_n_obj2coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/objects365/dfine_hgnetv2_n_obj2coco.yml -------------------------------------------------------------------------------- /configs/dfine/objects365/dfine_hgnetv2_n_obj365.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/objects365/dfine_hgnetv2_n_obj365.yml -------------------------------------------------------------------------------- /configs/dfine/objects365/dfine_hgnetv2_s_obj2coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/objects365/dfine_hgnetv2_s_obj2coco.yml -------------------------------------------------------------------------------- /configs/dfine/objects365/dfine_hgnetv2_s_obj365.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/objects365/dfine_hgnetv2_s_obj365.yml -------------------------------------------------------------------------------- /configs/dfine/objects365/dfine_hgnetv2_x_obj2coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/objects365/dfine_hgnetv2_x_obj2coco.yml -------------------------------------------------------------------------------- /configs/dfine/objects365/dfine_hgnetv2_x_obj365.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/dfine/objects365/dfine_hgnetv2_x_obj365.yml -------------------------------------------------------------------------------- /configs/runtime.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/configs/runtime.yml -------------------------------------------------------------------------------- /reference/convert_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/reference/convert_weight.py -------------------------------------------------------------------------------- /reference/safe_training.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/reference/safe_training.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/core/__init__.py -------------------------------------------------------------------------------- /src/core/_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/core/_config.py -------------------------------------------------------------------------------- /src/core/workspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/core/workspace.py -------------------------------------------------------------------------------- /src/core/yaml_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/core/yaml_config.py -------------------------------------------------------------------------------- /src/core/yaml_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/core/yaml_utils.py -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/__init__.py -------------------------------------------------------------------------------- /src/data/_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/_misc.py -------------------------------------------------------------------------------- /src/data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/dataloader.py -------------------------------------------------------------------------------- /src/data/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/dataset/__init__.py -------------------------------------------------------------------------------- /src/data/dataset/_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/dataset/_dataset.py -------------------------------------------------------------------------------- /src/data/dataset/cifar_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/dataset/cifar_dataset.py -------------------------------------------------------------------------------- /src/data/dataset/coco_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/dataset/coco_dataset.py -------------------------------------------------------------------------------- /src/data/dataset/coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/dataset/coco_eval.py -------------------------------------------------------------------------------- /src/data/dataset/coco_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/dataset/coco_utils.py -------------------------------------------------------------------------------- /src/data/dataset/voc_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/dataset/voc_detection.py -------------------------------------------------------------------------------- /src/data/dataset/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/dataset/voc_eval.py -------------------------------------------------------------------------------- /src/data/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/transforms/__init__.py -------------------------------------------------------------------------------- /src/data/transforms/_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/transforms/_transforms.py -------------------------------------------------------------------------------- /src/data/transforms/container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/transforms/container.py -------------------------------------------------------------------------------- /src/data/transforms/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/transforms/functional.py -------------------------------------------------------------------------------- /src/data/transforms/mosaic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/transforms/mosaic.py -------------------------------------------------------------------------------- /src/data/transforms/presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/data/transforms/presets.py -------------------------------------------------------------------------------- /src/misc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/misc/__init__.py -------------------------------------------------------------------------------- /src/misc/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/misc/box_ops.py -------------------------------------------------------------------------------- /src/misc/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/misc/dist_utils.py -------------------------------------------------------------------------------- /src/misc/lazy_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/misc/lazy_loader.py -------------------------------------------------------------------------------- /src/misc/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/misc/logger.py -------------------------------------------------------------------------------- /src/misc/profiler_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/misc/profiler_utils.py -------------------------------------------------------------------------------- /src/misc/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/misc/visualizer.py -------------------------------------------------------------------------------- /src/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/__init__.py -------------------------------------------------------------------------------- /src/nn/arch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/arch/__init__.py -------------------------------------------------------------------------------- /src/nn/arch/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/arch/classification.py -------------------------------------------------------------------------------- /src/nn/arch/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/arch/yolo.py -------------------------------------------------------------------------------- /src/nn/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/backbone/__init__.py -------------------------------------------------------------------------------- /src/nn/backbone/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/backbone/common.py -------------------------------------------------------------------------------- /src/nn/backbone/csp_darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/backbone/csp_darknet.py -------------------------------------------------------------------------------- /src/nn/backbone/csp_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/backbone/csp_resnet.py -------------------------------------------------------------------------------- /src/nn/backbone/hgnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/backbone/hgnetv2.py -------------------------------------------------------------------------------- /src/nn/backbone/presnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/backbone/presnet.py -------------------------------------------------------------------------------- /src/nn/backbone/test_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/backbone/test_resnet.py -------------------------------------------------------------------------------- /src/nn/backbone/timm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/backbone/timm_model.py -------------------------------------------------------------------------------- /src/nn/backbone/torchvision_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/backbone/torchvision_model.py -------------------------------------------------------------------------------- /src/nn/backbone/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/backbone/utils.py -------------------------------------------------------------------------------- /src/nn/criterion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/criterion/__init__.py -------------------------------------------------------------------------------- /src/nn/criterion/det_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/criterion/det_criterion.py -------------------------------------------------------------------------------- /src/nn/postprocessor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/postprocessor/__init__.py -------------------------------------------------------------------------------- /src/nn/postprocessor/box_revert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/postprocessor/box_revert.py -------------------------------------------------------------------------------- /src/nn/postprocessor/detr_postprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/postprocessor/detr_postprocessor.py -------------------------------------------------------------------------------- /src/nn/postprocessor/nms_postprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/nn/postprocessor/nms_postprocessor.py -------------------------------------------------------------------------------- /src/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/optim/__init__.py -------------------------------------------------------------------------------- /src/optim/amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/optim/amp.py -------------------------------------------------------------------------------- /src/optim/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/optim/ema.py -------------------------------------------------------------------------------- /src/optim/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/optim/optim.py -------------------------------------------------------------------------------- /src/optim/warmup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/optim/warmup.py -------------------------------------------------------------------------------- /src/solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/solver/__init__.py -------------------------------------------------------------------------------- /src/solver/_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/solver/_solver.py -------------------------------------------------------------------------------- /src/solver/clas_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/solver/clas_engine.py -------------------------------------------------------------------------------- /src/solver/clas_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/solver/clas_solver.py -------------------------------------------------------------------------------- /src/solver/det_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/solver/det_engine.py -------------------------------------------------------------------------------- /src/solver/det_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/solver/det_solver.py -------------------------------------------------------------------------------- /src/solver/validator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/solver/validator.py -------------------------------------------------------------------------------- /src/zoo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/__init__.py -------------------------------------------------------------------------------- /src/zoo/dfine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/__init__.py -------------------------------------------------------------------------------- /src/zoo/dfine/blog.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/blog.md -------------------------------------------------------------------------------- /src/zoo/dfine/blog_cn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/blog_cn.md -------------------------------------------------------------------------------- /src/zoo/dfine/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/box_ops.py -------------------------------------------------------------------------------- /src/zoo/dfine/denoising.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/denoising.py -------------------------------------------------------------------------------- /src/zoo/dfine/dfine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/dfine.py -------------------------------------------------------------------------------- /src/zoo/dfine/dfine_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/dfine_criterion.py -------------------------------------------------------------------------------- /src/zoo/dfine/dfine_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/dfine_decoder.py -------------------------------------------------------------------------------- /src/zoo/dfine/dfine_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/dfine_utils.py -------------------------------------------------------------------------------- /src/zoo/dfine/hybrid_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/hybrid_encoder.py -------------------------------------------------------------------------------- /src/zoo/dfine/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/matcher.py -------------------------------------------------------------------------------- /src/zoo/dfine/postprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/postprocessor.py -------------------------------------------------------------------------------- /src/zoo/dfine/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/src/zoo/dfine/utils.py -------------------------------------------------------------------------------- /tools/benchmark/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/benchmark/dataset.py -------------------------------------------------------------------------------- /tools/benchmark/get_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/benchmark/get_info.py -------------------------------------------------------------------------------- /tools/benchmark/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/benchmark/requirements.txt -------------------------------------------------------------------------------- /tools/benchmark/trt_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/benchmark/trt_benchmark.py -------------------------------------------------------------------------------- /tools/benchmark/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/benchmark/utils.py -------------------------------------------------------------------------------- /tools/dataset/remap_obj365.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/dataset/remap_obj365.py -------------------------------------------------------------------------------- /tools/dataset/resize_obj365.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/dataset/resize_obj365.py -------------------------------------------------------------------------------- /tools/deployment/export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/deployment/export_onnx.py -------------------------------------------------------------------------------- /tools/deployment/export_yolo_w_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/deployment/export_yolo_w_nms.py -------------------------------------------------------------------------------- /tools/inference/cppExample/onnx/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/inference/cppExample/onnx/CMakeLists.txt -------------------------------------------------------------------------------- /tools/inference/cppExample/onnx/onnxExample.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/inference/cppExample/onnx/onnxExample.cpp -------------------------------------------------------------------------------- /tools/inference/cppExample/openvino/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/inference/cppExample/openvino/CMakeLists.txt -------------------------------------------------------------------------------- /tools/inference/cppExample/openvino/openvinoExample.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/inference/cppExample/openvino/openvinoExample.cpp -------------------------------------------------------------------------------- /tools/inference/cppExample/trt/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/inference/cppExample/trt/CMakeLists.txt -------------------------------------------------------------------------------- /tools/inference/cppExample/trt/trtExample.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/inference/cppExample/trt/trtExample.cpp -------------------------------------------------------------------------------- /tools/inference/onnx_inf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/inference/onnx_inf.py -------------------------------------------------------------------------------- /tools/inference/openvino_inf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/inference/openvino_inf.py -------------------------------------------------------------------------------- /tools/inference/requirements.txt: -------------------------------------------------------------------------------- 1 | onnxruntime 2 | tensorrt 3 | openvino 4 | -------------------------------------------------------------------------------- /tools/inference/torch_inf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/inference/torch_inf.py -------------------------------------------------------------------------------- /tools/inference/trt_inf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/inference/trt_inf.py -------------------------------------------------------------------------------- /tools/visualization/fiftyone_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/tools/visualization/fiftyone_vis.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Peterande/D-FINE/HEAD/train.py --------------------------------------------------------------------------------