Merge README edits from HF
Browse files
README.md
CHANGED
|
@@ -31,15 +31,15 @@ pipeline_tag: image-classification
|
|
| 31 |
|
| 32 |
| Model | Params (M) | MACs (G) | Top-1 Acc (%) | Checkpoint (pth) |
|
| 33 |
|:---|:---:|:---:|:---:|:---:|
|
| 34 |
-
| GeoViG-Ti | 3.5 | 0.9 | 75.2 | [Download](https://huggingface.co/
|
| 35 |
-
| GeoViG-S | 5.0 | 1.2 | 77.5 | [Download](https://huggingface.co/
|
| 36 |
-
| GeoViG-M | 10.3 | 2.2 | 80.7 | [Download](https://huggingface.co/
|
| 37 |
-
| GeoViG-B | 19.7 | 4.5 | 82.4 | [Download](https://huggingface.co/
|
| 38 |
|
| 39 |
Trained for 300 epochs on 8Γ NVIDIA A100 GPUs, batch size 1024, AdamW optimizer (lr=5e-4, weight decay=0.05), cosine schedule with 5-epoch warmup. Augmentations: RandAugment, Mixup (p=0.8), CutMix (p=1.0).
|
| 40 |
|
| 41 |
-
For CoreML checkpoint, please check:
|
| 42 |
-
For IPA chechpoints, please check:
|
| 43 |
---
|
| 44 |
|
| 45 |
### π¦ Object Detection & Instance Segmentation β MS COCO 2017
|
|
@@ -48,8 +48,8 @@ Backbone used with **Mask R-CNN**, 1Γ schedule (12 epochs), pretrained on Image
|
|
| 48 |
|
| 49 |
| Backbone | Params (M) | Box AP | Box APβ
β | Box APββ
| Mask AP | Mask APβ
β | Mask APββ
| Checkpoint |
|
| 50 |
|:---|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|
|
| 51 |
-
| GeoViG-M | 10.3 | 40.7 | 62.4 | 44.1 | 37.7 | 59.6 | 40.5 | [Download](https://huggingface.co/
|
| 52 |
-
| GeoViG-B | 19.7 | 42.5 | 64.0 | 46.8 | 38.9 | 61.2 | 41.7 | [Download](https://huggingface.co/
|
| 53 |
|
| 54 |
---
|
| 55 |
|
|
@@ -59,13 +59,13 @@ Backbone used with **Mask R-CNN**, 1Γ schedule (12 epochs), pretrained on Image
|
|
| 59 |
|
| 60 |
| Model | Params (M) | mAP | Dice β | IoU β | Hausdorff Dist β | Checkpoint |
|
| 61 |
|:---|:---:|:---:|:---:|:---:|:---:|:---:|
|
| 62 |
-
| GeoViG-M | 29.57 | 0.990 | **0.945** | **0.909** | **12.94** | [Download](https://huggingface.co/
|
| 63 |
|
| 64 |
**Data Science Bowl 2018 β Nuclei Segmentation**
|
| 65 |
|
| 66 |
| Model | Params (M) | mAP | Dice β | IoU β | Hausdorff Dist β | Checkpoint |
|
| 67 |
|:---|:---:|:---:|:---:|:---:|:---:|:---:|
|
| 68 |
-
| GeoViG-M | 29.57 | 0.859 | **0.908** | **0.839** | **5.19** | [Download](https://huggingface.co/
|
| 69 |
|
| 70 |
---
|
| 71 |
|
|
|
|
| 31 |
|
| 32 |
| Model | Params (M) | MACs (G) | Top-1 Acc (%) | Checkpoint (pth) |
|
| 33 |
|:---|:---:|:---:|:---:|:---:|
|
| 34 |
+
| GeoViG-Ti | 3.5 | 0.9 | 75.2 | [Download](https://huggingface.co/OmarAlasqa/GeoViG/blob/main/pth/geovig_ti_5e4_8G_300_75_22/checkpoint.pth) |
|
| 35 |
+
| GeoViG-S | 5.0 | 1.2 | 77.5 | [Download](https://huggingface.co/OmarAlasqa/GeoViG/blob/main/pth/geovig_s_5e4_8G_300_77_48/checkpoint.pth) |
|
| 36 |
+
| GeoViG-M | 10.3 | 2.2 | 80.7 | [Download](https://huggingface.co/OmarAlasqa/GeoViG/blob/main/pth/geovig_m_5e4_8G_300_80_70/checkpoint.pth) |
|
| 37 |
+
| GeoViG-B | 19.7 | 4.5 | 82.4 | [Download](https://huggingface.co/OmarAlasqa/GeoViG/blob/main/pth/geovig_b_5e4_8G_300_82_38/checkpoint.pth) |
|
| 38 |
|
| 39 |
Trained for 300 epochs on 8Γ NVIDIA A100 GPUs, batch size 1024, AdamW optimizer (lr=5e-4, weight decay=0.05), cosine schedule with 5-epoch warmup. Augmentations: RandAugment, Mixup (p=0.8), CutMix (p=1.0).
|
| 40 |
|
| 41 |
+
For CoreML checkpoint, please check: https://huggingface.co/OmarAlasqa/GeoViG/tree/main/CoreML
|
| 42 |
+
For IPA chechpoints, please check: https://huggingface.co/OmarAlasqa/GeoViG/tree/main/IPA
|
| 43 |
---
|
| 44 |
|
| 45 |
### π¦ Object Detection & Instance Segmentation β MS COCO 2017
|
|
|
|
| 48 |
|
| 49 |
| Backbone | Params (M) | Box AP | Box APβ
β | Box APββ
| Mask AP | Mask APβ
β | Mask APββ
| Checkpoint |
|
| 50 |
|:---|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|
|
| 51 |
+
| GeoViG-M | 10.3 | 40.7 | 62.4 | 44.1 | 37.7 | 59.6 | 40.5 | [Download](https://huggingface.co/OmarAlasqa/GeoViG/blob/main/coco_det_seg_pth/geovig_m_det_seg/epoch_12.pth) |
|
| 52 |
+
| GeoViG-B | 19.7 | 42.5 | 64.0 | 46.8 | 38.9 | 61.2 | 41.7 | [Download](https://huggingface.co/OmarAlasqa/GeoViG/blob/main/coco_det_seg_pth/geovig_b_det_seg/epoch_12.pth) |
|
| 53 |
|
| 54 |
---
|
| 55 |
|
|
|
|
| 59 |
|
| 60 |
| Model | Params (M) | mAP | Dice β | IoU β | Hausdorff Dist β | Checkpoint |
|
| 61 |
|:---|:---:|:---:|:---:|:---:|:---:|:---:|
|
| 62 |
+
| GeoViG-M | 29.57 | 0.990 | **0.945** | **0.909** | **12.94** | [Download](https://huggingface.co/OmarAlasqa/GeoViG/blob/main/medical/kvasir_geovig_m/checkpoint.pth) |
|
| 63 |
|
| 64 |
**Data Science Bowl 2018 β Nuclei Segmentation**
|
| 65 |
|
| 66 |
| Model | Params (M) | mAP | Dice β | IoU β | Hausdorff Dist β | Checkpoint |
|
| 67 |
|:---|:---:|:---:|:---:|:---:|:---:|:---:|
|
| 68 |
+
| GeoViG-M | 29.57 | 0.859 | **0.908** | **0.839** | **5.19** | [Download](https://huggingface.co/OmarAlasqa/GeoViG/blob/main/medical/dsb_geovig_m/checkpoint.pth) |
|
| 69 |
|
| 70 |
---
|
| 71 |
|