SSP-DETR Model Checkpoints
This repository contains the SSP-DETR checkpoints reported in the paper: one training run with each of the random seeds 42, 43, and 44 on each dataset. Code: https://github.com/coolAning/SSP-DETR-Code.
| Dataset | Seed | Checkpoint | AP | AP75 | APS | SHA256 |
|---|---|---|---|---|---|---|
| PCB | 42 | ssp_detr_pcb_seed42.pth |
57.04 | 59.31 | 56.60 | f48bdfd5aea9e9c8203d3132877cfa02e28dc382647a03d143885f9b0502507a |
| PCB | 43 | ssp_detr_pcb_seed43.pth |
57.49 | 64.11 | 57.15 | 1d822f2c31902fdbe762dbd25c574e06cb914798199632587f7db480675d08b4 |
| PCB | 44 | ssp_detr_pcb_seed44.pth |
56.51 | 57.46 | 55.99 | 02a7db98445c230bf58d0cbf3b8178573157e2a7afc178dd05666893d910a2ab |
| CPS2D | 42 | ssp_detr_cps2d_seed42.pth |
23.14 | 17.99 | 19.89 | 93add214bf0542204408a532e57ef40a371fcfd007da265cfebd76c86e828735 |
| CPS2D | 43 | ssp_detr_cps2d_seed43.pth |
24.18 | 19.05 | 20.91 | 0d560ecee94fc784f76fd7c905869cf9fa24f21cf4efafb466e9b607f6319c36 |
| CPS2D | 44 | ssp_detr_cps2d_seed44.pth |
23.38 | 18.61 | 20.21 | 156d25de016404c7317c85f43a8be8c99c03e56fabc2f8142df21b3ba800abdf |
| CPH2D | 42 | ssp_detr_cph_seed42.pth |
31.27 | 24.81 | 27.98 | 118490f50ec068f1111aa009db80adb9680d3b68b111301722d6f71d7b6e9b13 |
| CPH2D | 43 | ssp_detr_cph_seed43.pth |
29.70 | 32.21 | 23.84 | 3bb9c3ae177662707c9a62101b720fa9819f1e7b444d305fea850d2a0492d7e8 |
| CPH2D | 44 | ssp_detr_cph_seed44.pth |
32.93 | 34.63 | 25.95 | 3e51dfd866e54cae4659a906463bc03048049bcd78b109c326444c89f38a40bd |
Mean ± sample standard deviation over the three seeds:
| Dataset | AP | AP75 | APS |
|---|---|---|---|
| PCB | 57.0 ± 0.5 | 60.3 ± 3.4 | 56.6 ± 0.6 |
| CPS2D | 23.6 ± 0.5 | 18.6 ± 0.5 | 20.3 ± 0.5 |
| CPH2D | 31.3 ± 1.6 | 30.6 ± 5.1 | 25.9 ± 2.1 |
The AP values are obtained with scripts/eval.sh from the code repository, for example DATASET=pcb SEED=43 bash scripts/eval.sh. Each checkpoint stores the model state dictionary under the model key.