leowizard commited on
Commit
4ee7238
·
verified ·
1 Parent(s): 56ebf1a

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