Add pipeline tag and code link
#1
by
nielsr HF Staff - opened
README.md
CHANGED
|
@@ -1,8 +1,10 @@
|
|
| 1 |
---
|
| 2 |
-
license: mit
|
| 3 |
language:
|
| 4 |
- en
|
|
|
|
|
|
|
| 5 |
---
|
|
|
|
| 6 |
This is the official repo where we would release the weights of [UniRGB-IR](https://arxiv.org/abs/2404.17360).
|
| 7 |
|
| 8 |
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).
|
|
@@ -11,3 +13,4 @@ We have also uploaded our pretrained weight `vitb_IN1k_mae_coco_cascade-mask-rcn
|
|
| 11 |
|----------------|-----------|---------------------------------------|-----------------------------------------------------|
|
| 12 |
| 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 |
|
| 13 |
|
|
|
|
|
|
| 1 |
---
|
|
|
|
| 2 |
language:
|
| 3 |
- en
|
| 4 |
+
license: mit
|
| 5 |
+
pipeline_tag: image-to-image
|
| 6 |
---
|
| 7 |
+
|
| 8 |
This is the official repo where we would release the weights of [UniRGB-IR](https://arxiv.org/abs/2404.17360).
|
| 9 |
|
| 10 |
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).
|
|
|
|
| 13 |
|----------------|-----------|---------------------------------------|-----------------------------------------------------|
|
| 14 |
| 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 |
|
| 15 |
|
| 16 |
+
Code: https://github.com/MaoxunYuan/UniRGB-IR
|