This is the official repo where we would release the weights of UniRGB-IR.

We have also uploaded our pretrained weight vitb_IN1k_mae_coco_cascade-mask-rcnn_224x224_noClsToken_noRel.pth, which is converted from from ViTDet.

Task Dataset Pretrained Weight Metrics
Object Detection FLIR_align flir_1024_bsz8x1_iter_42500.pth coco/bbox_mAP: 0.4510
coco/bbox_mAP_50: 0.8270
coco/bbox_mAP_75: 0.4270
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.