OmarAlasqa commited on
Commit
9d4f2fd
Β·
2 Parent(s): 1e119e8 8e3e846

Merge README edits from HF

Browse files
Files changed (1) hide show
  1. README.md +10 -10
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/TODO/geovig/resolve/main/pth/geovig_ti.pth) |
35
- | GeoViG-S | 5.0 | 1.2 | 77.5 | [Download](https://huggingface.co/TODO/geovig/resolve/main/pth/geovig_s.pth) |
36
- | GeoViG-M | 10.3 | 2.2 | 80.7 | [Download](https://huggingface.co/TODO/geovig/resolve/main/pth/geovig_m.pth) |
37
- | GeoViG-B | 19.7 | 4.5 | 82.4 | [Download](https://huggingface.co/TODO/geovig/resolve/main/pth/geovig_b.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:
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/TODO/geovig/resolve/main/detection/geovig_m_coco.pth) |
52
- | GeoViG-B | 19.7 | 42.5 | 64.0 | 46.8 | 38.9 | 61.2 | 41.7 | [Download](https://huggingface.co/TODO/geovig/resolve/main/detection/geovig_b_coco.pth) |
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/TODO/geovig/resolve/main/medical/geovig_m_kvasir.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/TODO/geovig/resolve/main/medical/geovig_m_dsb2018.pth) |
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