├── .gitignore ├── LICENSE ├── README.md ├── configs ├── hyp.no-augmentation.yaml ├── hyp.scratch-high.yaml ├── hyp.scratch-low.yaml ├── hyp.scratch-low_v2.yaml ├── hyp.scratch-low_v3.yaml ├── hyp.scratch-med.yaml └── hyp.scratch-test.yaml ├── dataset ├── base_dataset.py ├── llvip.yaml ├── m3fd.yaml ├── result.yaml ├── rgbt_dataset.py ├── test.yaml └── utokyo.yaml ├── debug.py ├── detect.py ├── detect.sh ├── export.py ├── model ├── backbone.py ├── common.py ├── decoder.py ├── frame.py ├── fuseblock.py ├── head.py ├── mamba │ ├── light_mamba.py │ ├── mamba.py │ └── vmamba.py ├── neck.py ├── replknet.py └── unireplknet.py ├── requirements.txt ├── requirements.txt.bak ├── test_imgs ├── 1686403336-rgb.png └── 1686403336-t.png ├── tools ├── build_engine_from_onnx.py ├── build_engine_from_onnx_v2.py ├── eval_quantize.py ├── onnx_gs.py ├── ppq_quantization.py ├── ppq_quantizer.py └── trt_infer.py ├── train.py ├── train.sh ├── utils ├── anchor.py ├── autobatch.py ├── callbacks.py ├── coco │ ├── coco_eval.py │ ├── coco_utils.py │ ├── engine.py │ └── transforms.py ├── general.py ├── loggers │ ├── __init__.py │ ├── clearml │ │ ├── README.md │ │ ├── __init__.py │ │ ├── clearml_utils.py │ │ └── hpo.py │ ├── comet │ │ ├── README.md │ │ ├── __init__.py │ │ ├── comet_utils.py │ │ └── hpo.py │ └── wandb │ │ ├── __init__.py │ │ └── wandb_utils.py ├── loss.py ├── metrics.py ├── plots.py └── torch_utils.py ├── val.py └── val.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/README.md -------------------------------------------------------------------------------- /configs/hyp.no-augmentation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/configs/hyp.no-augmentation.yaml -------------------------------------------------------------------------------- /configs/hyp.scratch-high.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/configs/hyp.scratch-high.yaml -------------------------------------------------------------------------------- /configs/hyp.scratch-low.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/configs/hyp.scratch-low.yaml -------------------------------------------------------------------------------- /configs/hyp.scratch-low_v2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/configs/hyp.scratch-low_v2.yaml -------------------------------------------------------------------------------- /configs/hyp.scratch-low_v3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/configs/hyp.scratch-low_v3.yaml -------------------------------------------------------------------------------- /configs/hyp.scratch-med.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/configs/hyp.scratch-med.yaml -------------------------------------------------------------------------------- /configs/hyp.scratch-test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/configs/hyp.scratch-test.yaml -------------------------------------------------------------------------------- /dataset/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/dataset/base_dataset.py -------------------------------------------------------------------------------- /dataset/llvip.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/dataset/llvip.yaml -------------------------------------------------------------------------------- /dataset/m3fd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/dataset/m3fd.yaml -------------------------------------------------------------------------------- /dataset/result.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/dataset/result.yaml -------------------------------------------------------------------------------- /dataset/rgbt_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/dataset/rgbt_dataset.py -------------------------------------------------------------------------------- /dataset/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/dataset/test.yaml -------------------------------------------------------------------------------- /dataset/utokyo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/dataset/utokyo.yaml -------------------------------------------------------------------------------- /debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/debug.py -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/detect.py -------------------------------------------------------------------------------- /detect.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/detect.sh -------------------------------------------------------------------------------- /export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/export.py -------------------------------------------------------------------------------- /model/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/model/backbone.py -------------------------------------------------------------------------------- /model/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/model/common.py -------------------------------------------------------------------------------- /model/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/model/decoder.py -------------------------------------------------------------------------------- /model/frame.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/model/frame.py -------------------------------------------------------------------------------- /model/fuseblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/model/fuseblock.py -------------------------------------------------------------------------------- /model/head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/model/head.py -------------------------------------------------------------------------------- /model/mamba/light_mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/model/mamba/light_mamba.py -------------------------------------------------------------------------------- /model/mamba/mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/model/mamba/mamba.py -------------------------------------------------------------------------------- /model/mamba/vmamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/model/mamba/vmamba.py -------------------------------------------------------------------------------- /model/neck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/model/neck.py -------------------------------------------------------------------------------- /model/replknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/model/replknet.py -------------------------------------------------------------------------------- /model/unireplknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/model/unireplknet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements.txt.bak: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/requirements.txt.bak -------------------------------------------------------------------------------- /test_imgs/1686403336-rgb.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/test_imgs/1686403336-rgb.png -------------------------------------------------------------------------------- /test_imgs/1686403336-t.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/test_imgs/1686403336-t.png -------------------------------------------------------------------------------- /tools/build_engine_from_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/tools/build_engine_from_onnx.py -------------------------------------------------------------------------------- /tools/build_engine_from_onnx_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/tools/build_engine_from_onnx_v2.py -------------------------------------------------------------------------------- /tools/eval_quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/tools/eval_quantize.py -------------------------------------------------------------------------------- /tools/onnx_gs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/tools/onnx_gs.py -------------------------------------------------------------------------------- /tools/ppq_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/tools/ppq_quantization.py -------------------------------------------------------------------------------- /tools/ppq_quantizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/tools/ppq_quantizer.py -------------------------------------------------------------------------------- /tools/trt_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/tools/trt_infer.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/train.sh -------------------------------------------------------------------------------- /utils/anchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/anchor.py -------------------------------------------------------------------------------- /utils/autobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/autobatch.py -------------------------------------------------------------------------------- /utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/callbacks.py -------------------------------------------------------------------------------- /utils/coco/coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/coco/coco_eval.py -------------------------------------------------------------------------------- /utils/coco/coco_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/coco/coco_utils.py -------------------------------------------------------------------------------- /utils/coco/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/coco/engine.py -------------------------------------------------------------------------------- /utils/coco/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/coco/transforms.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/loggers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/loggers/__init__.py -------------------------------------------------------------------------------- /utils/loggers/clearml/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/loggers/clearml/README.md -------------------------------------------------------------------------------- /utils/loggers/clearml/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/loggers/clearml/clearml_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/loggers/clearml/clearml_utils.py -------------------------------------------------------------------------------- /utils/loggers/clearml/hpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/loggers/clearml/hpo.py -------------------------------------------------------------------------------- /utils/loggers/comet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/loggers/comet/README.md -------------------------------------------------------------------------------- /utils/loggers/comet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/loggers/comet/__init__.py -------------------------------------------------------------------------------- /utils/loggers/comet/comet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/loggers/comet/comet_utils.py -------------------------------------------------------------------------------- /utils/loggers/comet/hpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/loggers/comet/hpo.py -------------------------------------------------------------------------------- /utils/loggers/wandb/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/loggers/wandb/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/loggers/wandb/wandb_utils.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/val.py -------------------------------------------------------------------------------- /val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CrazyThermos/rgbt-detection/HEAD/val.sh --------------------------------------------------------------------------------