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 |
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.