background-mixup / README.md
leowizard's picture
Add baseline and BGM PIDray checkpoints
4ee7238 verified
|
Raw History Blame Contribute Delete
1.16 kB
metadata
language: en
pipeline_tag: object-detection
tags:
  - x-ray
  - bgm
  - pidray
  - detrex

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.