| --- |
| license: other |
| library_name: pytorch |
| tags: |
| - object-detection |
| - aerial-object-detection |
| - small-object-detection |
| - visdrone |
| - rtdetrv2 |
| - dfine |
| - tcsr |
| --- |
| |
| # TCSR VisDrone Checkpoints |
|
|
| This repository contains the public VisDrone checkpoints for **Shallow-Response Calibration for Real-Time Aerial Small Object Detection**. The implementation, configurations, training commands, and evaluation tools are maintained in the [TCSR GitHub repository](https://github.com/txu8847-wq/TCSR). |
|
|
| Because of storage and long-term maintenance constraints, the hosted checkpoint collection is limited to VisDrone. The AI-TOD-v2 experiment configurations remain available in the GitHub repository. |
|
|
| ## Available Checkpoints |
|
|
| | Framework | Backbone | Variant | VisDrone AP | AP gain | File | |
| |:--|:--|:--|--:|--:|:--| |
| | D-FINE | HGNetv2-S | Baseline | 27.56 | - | `dfine_hgnetv2_s_visdrone_baseline_ap=27.56.pth` | |
| | D-FINE | HGNetv2-S | Direct | 28.51 | +0.95 | `dfine_hgnetv2_s_visdrone_detection_direct_hard_ap=28.51.pth` | |
| | D-FINE | HGNetv2-S | TCSR-Light | 29.11 | +1.55 | `dfine_hgnetv2_s_visdrone_TCSR_light=64_ap=29.11.pth` | |
| | D-FINE | HGNetv2-S | TCSR-Hard | 30.71 | +3.15 | `dfine_hgnetv2_s_visdrone_detection_TCSR_hard_ap=30.71.pth` | |
| | RT-DETRv2 | HGNetv2-X | Baseline | 26.52 | - | `rtdetrv2_hgnetv2_x_visdrone_baseline_72e_seed0_ap=26.52.pth` | |
| | RT-DETRv2 | HGNetv2-X | SE-Direct | 28.07 | +1.55 | `rtdetrv2_hgnetv2_x_visdrone_se_direct_72e_seed0_ap=28.07.pth` | |
| | RT-DETRv2 | HGNetv2-X | TCSR-Hard (deterministic) | 29.37 | +2.85 | `rtdetrv2_hgnetv2_x_visdrone_tcsr_det_72e_seed=0_ap=29.37.pth` | |
| | RT-DETRv2 | HGNetv2-X | TCSR-Hard | 29.30 | +2.78 | `rtdetrv2_hgnetv2_x_visdrone_TCSR_72e_seed0_AP=29.30.pth` | |
| | RT-DETRv2 | ResNet-18 | Baseline | 29.40 | - | `rtdetrv2_r18vd_visdrone_baseline_200e_seed0_ap=29.40.pth` | |
| | RT-DETRv2 | ResNet-18 | Direct | 29.76 | +0.36 | `rtdetrv2_r18vd_visdrone_direct_200e_seed0_ap=29.76.pth` | |
| | RT-DETRv2 | ResNet-18 | TCSR-Hard | 31.17 | +1.77 | `rtdetrv2_r18vd_200e_visdrone_TCSR_hard=256_ap=31.17.pth` | |
|
|
| The table lists the checkpoint files currently hosted in this repository. The revised paper reports the complete RT-DETRv2/ResNet-18 sequence: Baseline 29.40 AP, Direct 29.76 AP (+0.36), TCSR-Light 30.13 AP (+0.73), and TCSR-Hard 31.17 AP (+1.77). Results for additional variants are summarized in the [TCSR GitHub repository](https://github.com/txu8847-wq/TCSR). |
|
|
| For RT-DETRv2/HGNetv2-X, SE-Direct is the matched global channel-recalibration control. It uses standard squeeze-and-excitation after `C2` alignment, a reduction ratio of 16, no stochastic response exposure, the 72-epoch schedule, automatic mixed precision, and seed `0`. Its matching configuration is `RT-DETR/rtdetrv2_pytorch/configs/rtdetrv2/rtdetrv2_hgnetv2_x_visdrone_se_direct.yml` in the GitHub repository. |
|
|
| The deterministic TCSR-Hard checkpoint retains the complete 384-channel local TCSR transformation but disables stochastic response exposure (`tcsr_exposure_std: 0.0`). It uses the same 72-epoch HGNetv2-X setup, automatic mixed precision, and seed `0`; its matching GitHub configuration is `RT-DETR/rtdetrv2_pytorch/configs/rtdetrv2/rtdetrv2_hgnetv2_x_visdrone_tcsr_det.yml`. |
|
|
| ## Usage |
|
|
| Download one checkpoint with the Hugging Face CLI: |
|
|
| ```bash |
| hf download wodanile/TCSR <checkpoint-name>.pth --local-dir ./checkpoints |
| ``` |
|
|
| Refer to the GitHub repository for the matching configuration and use the framework's evaluation or resume option when loading a trained VisDrone checkpoint. The RT-DETRv2 and D-FINE loaders include compatibility remapping for checkpoints produced before the public TCSR namespace cleanup. |
|
|
| ## Important Notes |
|
|
| - These are PyTorch checkpoint files. Load checkpoints only from sources you trust. |
| - Checkpoint filenames record the framework, backbone, variant, schedule or seed where applicable, and reported validation AP. |
| - The COCO initialization checkpoint `rtdetrv2_r18vd_120e_coco_rerun_48.1.pth` is not part of this repository. |
|
|
