BGM: Background Mixup for X-Ray Prohibited Items Detection

Paper | Code

Baseline and BGM-trained Mr. DETR-Align R50 checkpoints for PIDray evaluation.

Model Positive AP Easy AP Hard AP Hidden AP
Baseline 74.8 78.9 74.3 52.7
BGM 76.8 80.4 76.3 56.1

These are recorded experiment results. The separated evaluation entry point has not yet been rerun on CUDA.

Files

  • output/mr_detr_align_r50_12ep/model_final.pth: baseline.
  • output/mr_detr_align_r50_12ep_bgm/model_final.pth: BGM.
  • CHECKPOINTS.json: checkpoint sizes and SHA-256 hashes.

Download the checkpoints into the matching paths in the code repository and follow its evaluation commands. Evaluation requires PIDray and a compatible Linux CUDA environment. Code repository access currently requires authorization.

The release provides checkpoint evaluation, without training reproduction, multi-seed results or separate CPM/SPM ablations.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Paper for leowizard/background-mixup