---
license: mit
language:
- en
---
This is the official repo where we would release the weights of [UniRGB-IR](https://arxiv.org/abs/2404.17360). 

We have also uploaded our pretrained weight `vitb_IN1k_mae_coco_cascade-mask-rcnn_224x224_noClsToken_noRel.pth`, which is converted from from [ViTDet](https://github.com/facebookresearch/detectron2/tree/main/projects/ViTDet#coco).

| Task           | Dataset   | Pretrained Weight                     | Metrics                                             |
|----------------|-----------|---------------------------------------|-----------------------------------------------------|
| Object Detection | FLIR_align | flir_1024_bsz8x1_iter_42500.pth       | coco/bbox_mAP: 0.4510<br>coco/bbox_mAP_50: 0.8270<br>coco/bbox_mAP_75: 0.4270 |