Upload folder using huggingface_hub

README.md

@@ -0,0 +1,55 @@
+---
+language: en
+license: mit
+tags:
+- fundus
+- diabetic retinopathy
+- classification
+datasets:
+- APTOS
+- EYEPACS
+- IDRID
+- DDR
+library: timm
+model-index:
+- name: convnext_small
+  results:
+  - task:
+      type: image-classification
+    dataset:
+      name: EYEPACS
+      type: EYEPACS
+    metrics:
+    - type: kappa
+      value: 0.789091944694519
+      name: Quadratic Kappa
+  - task:
+      type: image-classification
+    dataset:
+      name: IDRID
+      type: IDRID
+    metrics:
+    - type: kappa
+      value: 0.7588151097297668
+      name: Quadratic Kappa
+  - task:
+      type: image-classification
+    dataset:
+      name: DDR
+      type: DDR
+    metrics:
+    - type: kappa
+      value: 0.8041317462921143
+      name: Quadratic Kappa
+---
+# Fundus DR Grading
+
+[![Rye](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/rye/main/artwork/badge.json)](https://rye-up.com)
+[![PyTorch](https://img.shields.io/badge/PyTorch-ee4c2c?logo=pytorch&logoColor=white)](https://pytorch.org/docs/stable/index.html)
+[![Lightning](https://img.shields.io/badge/Lightning-792ee5?logo=lightning&logoColor=white)](https://lightning.ai/docs/pytorch/stable/)
+
+## Description
+
+This project aims to evaluate the performance of different models for the classification of diabetic retinopathy (DR) in fundus images. The reported perfomance metrics are not always consistent in the literature. Our goal is to provide a fair comparison between different models using the same datasets and evaluation protocol.
+
+

config.json

@@ -0,0 +1,41 @@
+{
+  "architecture": "convnext_small",
+  "num_classes": 1,
+  "num_features": 768,
+  "pretrained_cfg": {
+    "tag": "in12k_ft_in1k",
+    "custom_load": false,
+    "input_size": [
+      3,
+      224,
+      224
+    ],
+    "test_input_size": [
+      3,
+      288,
+      288
+    ],
+    "fixed_input_size": false,
+    "interpolation": "bicubic",
+    "crop_pct": 0.95,
+    "test_crop_pct": 1.0,
+    "crop_mode": "center",
+    "mean": [
+      0.485,
+      0.456,
+      0.406
+    ],
+    "std": [
+      0.229,
+      0.224,
+      0.225
+    ],
+    "num_classes": 1000,
+    "pool_size": [
+      7,
+      7
+    ],
+    "first_conv": "stem.0",
+    "classifier": "head.fc"
+  }
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0bae1cc619f806238c8c28fab477e2512a2506ce3674435f03b21e54310bbfdc
+size 197855900

preprocessor_config.json

@@ -0,0 +1,15 @@
+{
+  "do_normalize": true,
+  "do_resize": true,
+  "image_mean": [
+    0.485,
+    0.456,
+    0.406
+  ],
+  "image_std": [
+    0.229,
+    0.224,
+    0.225
+  ],
+  "size": 224
+}

preprocessor_config.json~

@@ -0,0 +1,15 @@
+{
+  "do_normalize": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "size": 224
+}

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c2787e09865e03dac5a35cfcf95fe35f77e4fd010224dce3754f39a7cd907a87
+size 197951722