Add baseline and BGM PIDray checkpoints
Browse files
CHECKPOINTS.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"output/mr_detr_align_r50_12ep/model_final.pth": {
|
| 3 |
+
"sha256": "3750f1788ccd8c798b3a3000c034fa51f2d3138b7125230f3773595b474489c7",
|
| 4 |
+
"bytes": 649573496
|
| 5 |
+
},
|
| 6 |
+
"output/mr_detr_align_r50_12ep_bgm/model_final.pth": {
|
| 7 |
+
"sha256": "9d5c9d3958e98aa64a32268c22b5628c304336a2aef5c16a92c6239932341a9e",
|
| 8 |
+
"bytes": 649573496
|
| 9 |
+
}
|
| 10 |
+
}
|
README.md
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: en
|
| 3 |
+
pipeline_tag: object-detection
|
| 4 |
+
tags:
|
| 5 |
+
- x-ray
|
| 6 |
+
- bgm
|
| 7 |
+
- pidray
|
| 8 |
+
- detrex
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# BGM: Background Mixup for X-Ray Prohibited Items Detection
|
| 12 |
+
|
| 13 |
+
[Paper](https://arxiv.org/abs/2412.00460) | [Code](https://github.com/WiZard-Leo/BackgroundMixup)
|
| 14 |
+
|
| 15 |
+
Baseline and BGM-trained Mr. DETR-Align R50 checkpoints for PIDray evaluation.
|
| 16 |
+
|
| 17 |
+
| Model | Positive AP | Easy AP | Hard AP | Hidden AP |
|
| 18 |
+
| --- | ---: | ---: | ---: | ---: |
|
| 19 |
+
| Baseline | 74.8 | 78.9 | 74.3 | 52.7 |
|
| 20 |
+
| BGM | 76.8 | 80.4 | 76.3 | 56.1 |
|
| 21 |
+
|
| 22 |
+
These are recorded experiment results. The separated evaluation entry point has not yet been rerun on CUDA.
|
| 23 |
+
|
| 24 |
+
## Files
|
| 25 |
+
|
| 26 |
+
- `output/mr_detr_align_r50_12ep/model_final.pth`: baseline.
|
| 27 |
+
- `output/mr_detr_align_r50_12ep_bgm/model_final.pth`: BGM.
|
| 28 |
+
- `CHECKPOINTS.json`: checkpoint sizes and SHA-256 hashes.
|
| 29 |
+
|
| 30 |
+
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.
|
| 31 |
+
|
| 32 |
+
The release provides checkpoint evaluation, without training reproduction, multi-seed results or separate CPM/SPM ablations.
|
output/mr_detr_align_r50_12ep/model_final.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3750f1788ccd8c798b3a3000c034fa51f2d3138b7125230f3773595b474489c7
|
| 3 |
+
size 649573496
|
output/mr_detr_align_r50_12ep_bgm/model_final.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9d5c9d3958e98aa64a32268c22b5628c304336a2aef5c16a92c6239932341a9e
|
| 3 |
+
size 649573496
|