background-mixup / README.md
leowizard's picture
Add baseline and BGM PIDray checkpoints
4ee7238 verified
|
Raw History Blame Contribute Delete
1.16 kB
---
language: en
pipeline_tag: object-detection
tags:
- x-ray
- bgm
- pidray
- detrex
---
# BGM: Background Mixup for X-Ray Prohibited Items Detection
[Paper](https://arxiv.org/abs/2412.00460) | [Code](https://github.com/WiZard-Leo/BackgroundMixup)
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.