Upload 43 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+predictions/2-10-original_image-4-PDR-0.438873.jpg filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,3 +1,32 @@
----
-license: cc-by-4.0
----
+# eyepacs-2015-2019-12-12-2023-093828/-12-12-2023-093828
+## Description
+Fine-tuned vit_large_patch16 model for ./models/eyepacs-2015-2019-12-12-2023-093828/
+
+## Use Cases
+- RETfound validation
+- Diabetic screening
+
+## Limitations
+- for research/education use only
+- not for clinical use
+
+## Ethics
+- Ethics 1
+- Ethics 2
+
+## Training Data
+6999 images from 5 classes
+
+## Training Procedure
+Fine-tuned for 50 epochs with batch size 10 and base learning rate 0.005
+
+## Intended Use
+Intended for use with eyepacs-2015-2019 dataset
+
+## Authors
+- Author 1
+- Author 2
+
+## References
+- Reference 1
+- Reference 2

_metrics_test.csv

@@ -0,0 +1 @@
+0.8507338225483656,0.6355602006688963,0.904512093411176,0.6324170234786031,0.9023067002137605,0.7223458664156602,0.631019392816882,0.5398242898795336,0.7276 (0.8580)

_metrics_val.csv

@@ -0,0 +1,50 @@
+0.7396666666666667,0.32166666666666666,0.8441666666666666,0.4155102654822791,0.7177366666666667,0.43496334773829126,0.2815057233352979,0.18651114071770006,1.4926 (1.5606)
+0.785,0.49833333333333335,0.8566666666666667,0.4727449340849328,0.8054355555555557,0.541647897465482,0.4603150388364957,0.3453491564008806,0.9357 (1.2812)
+0.8079999999999999,0.5316666666666667,0.8770833333333333,0.5200309589873396,0.8434700000000002,0.6016818433059261,0.5172845410645764,0.4048477532140208,0.8511 (1.1490)
+0.8009999999999999,0.49833333333333335,0.8766666666666667,0.5213628538690729,0.8528072222222223,0.6149140540384037,0.4854204385186067,0.38019926936146475,0.6356 (1.0941)
+0.8235,0.5616666666666666,0.8889583333333334,0.5746040487675187,0.8681294444444445,0.6408272255793718,0.550536567978012,0.45501570246320366,0.7634 (0.9974)
+0.8295,0.5700000000000001,0.894375,0.6105422877145438,0.8681138888888889,0.6352921495699688,0.5627177401492651,0.4783302333944587,0.8982 (0.9966)
+0.8311666666666667,0.62,0.8839583333333333,0.5960430215891891,0.8742849999999999,0.6505521110986071,0.5802927892812847,0.49673063060252975,0.6032 (0.9704)
+0.8286666666666667,0.6025,0.8852083333333334,0.5924092762657743,0.8768344444444445,0.6572677623771228,0.5857287963348088,0.48720784520606986,0.7876 (0.9704)
+0.8153333333333334,0.605,0.8679166666666667,0.5588867374210418,0.8678283333333333,0.643939642748423,0.5681512458705702,0.46298254239619097,0.6349 (1.0064)
+0.8313333333333333,0.605,0.8879166666666667,0.5997077183781594,0.8785572222222221,0.6587068884339569,0.596049925265322,0.4947424818461614,0.6414 (0.9608)
+0.8318333333333333,0.6166666666666667,0.885625,0.6105267991503277,0.8837455555555556,0.6753337690352369,0.6034315909457433,0.5050515153279728,0.6860 (0.9525)
+0.8291666666666667,0.6166666666666667,0.8822916666666666,0.6041154518758319,0.8795086111111111,0.657507272072271,0.6000578128219476,0.49991655138366986,0.7597 (0.9605)
+0.8268333333333333,0.6066666666666667,0.881875,0.5872940529103792,0.8839616666666668,0.6743105479597833,0.587237032084029,0.48560980826624267,0.6009 (0.9496)
+0.8346666666666667,0.6125,0.8902083333333334,0.6028116629988616,0.8837944444444445,0.6731118223122695,0.5948868830893859,0.5018601547490996,0.6604 (0.9393)
+0.8320000000000001,0.6233333333333334,0.8841666666666667,0.6022167452882015,0.8833383333333333,0.670574189189177,0.5999834790542142,0.504084989490353,0.7037 (0.9472)
+0.8435,0.5791666666666667,0.9095833333333333,0.6428600472948067,0.8838194444444444,0.6752147942825031,0.6006213193679445,0.5110349344658636,0.7593 (0.9509)
+0.8465,0.5358333333333334,0.9241666666666668,0.6422629144587961,0.8827777777777778,0.6686500008821878,0.575631852247975,0.4923395661847537,0.6399 (0.9492)
+0.8366666666666667,0.6308333333333334,0.8881249999999999,0.5913667642317447,0.8843650000000001,0.6738810108851558,0.6086138427614142,0.5075412891289571,0.4916 (0.9345)
+0.8406666666666666,0.6175,0.8964583333333334,0.6185460750575972,0.8849733333333333,0.6767549330417341,0.6106722581714857,0.5163175055736758,0.6605 (0.9180)
+0.8455,0.6141666666666666,0.9033333333333333,0.6291093708045147,0.8906244444444444,0.6882246297423366,0.6178608092580936,0.5240009281809721,0.6464 (0.9142)
+0.845,0.5958333333333333,0.9072916666666667,0.6166609261763203,0.890165,0.6864218050001634,0.6033247136210436,0.5093927254235675,0.5351 (0.9041)
+0.8443333333333334,0.6066666666666667,0.90375,0.6233707944756348,0.8878877777777777,0.6808123990311993,0.6074057734825206,0.516459180352726,0.6059 (0.9252)
+0.8463333333333334,0.6233333333333333,0.9020833333333333,0.6337785045206334,0.8922927777777778,0.6924827103251519,0.6241636663201013,0.5309794974590523,0.6802 (0.9003)
+0.851,0.6100000000000001,0.91125,0.6428241183011899,0.8893611111111112,0.6853722184399469,0.6223665286139346,0.5326800167372638,0.6876 (0.9091)
+0.8503333333333334,0.6125,0.9097916666666666,0.6313934584936549,0.8938883333333333,0.6938507049789355,0.6195661545320996,0.5283093872070703,0.5640 (0.8906)
+0.8488333333333334,0.6533333333333333,0.8977083333333333,0.6331892915863784,0.8903433333333334,0.6880678870199703,0.6380961111535435,0.5468078030282106,0.6896 (0.9017)
+0.8431666666666666,0.6491666666666667,0.8916666666666667,0.6177735887712881,0.8899750000000001,0.692424414018072,0.625241765050023,0.5328545257757492,0.7334 (0.9236)
+0.8518333333333333,0.6491666666666667,0.9025000000000001,0.6391155613418853,0.8901327777777777,0.6897161793904231,0.6406928485666882,0.5501399453161082,0.6350 (0.9140)
+0.8521666666666667,0.6308333333333334,0.9075,0.6356674748821192,0.8915683333333332,0.6934121482865411,0.6277528650500683,0.5400793896849517,0.5771 (0.9104)
+0.8541666666666666,0.6483333333333334,0.905625,0.6477613871255534,0.8941672222222221,0.7035708779701648,0.6405167226659829,0.5551238289497925,0.6943 (0.8963)
+0.8553333333333333,0.66,0.9041666666666667,0.6462571541786646,0.8949955555555557,0.7036580350270457,0.6478583455156843,0.5610852716669639,0.6557 (0.8941)
+0.8531666666666666,0.6475,0.9045833333333333,0.6420434037657158,0.8940972222222221,0.7011105491166856,0.640356560449189,0.5516719429967534,0.6034 (0.9025)
+0.8555,0.6383333333333333,0.9097916666666666,0.6461616255164375,0.8940566666666665,0.7001598369811075,0.6392614843934155,0.5513118589638941,0.6238 (0.9022)
+0.8583333333333333,0.6308333333333334,0.9152083333333333,0.6552846455639401,0.8966155555555556,0.7033514166962455,0.6390881303344973,0.5541187387439755,0.6613 (0.8851)
+0.8556666666666667,0.6333333333333333,0.91125,0.6485648520800781,0.8937261111111111,0.6937738548244913,0.637346083708794,0.5501227052363465,0.6437 (0.8974)
+0.8531666666666666,0.6341666666666667,0.9079166666666667,0.6429381276320052,0.8962033333333332,0.702170819304806,0.6357426965626113,0.5459808599682173,0.6809 (0.8839)
+0.8541666666666667,0.615,0.9139583333333333,0.6524310234505474,0.8967711111111111,0.703743879943718,0.6287841594462039,0.5418767792622303,0.7010 (0.8892)
+0.8566666666666667,0.6391666666666667,0.9110416666666667,0.6483877817926869,0.8953127777777778,0.7015650828260644,0.6409212706273046,0.5536709473732793,0.6628 (0.8925)
+0.857,0.6333333333333333,0.9129166666666666,0.6521682055956249,0.8965905555555554,0.7032947798759835,0.6385191426605538,0.5527797802872706,0.6951 (0.8872)
+0.8581666666666666,0.6383333333333334,0.913125,0.6510842438121972,0.897793888888889,0.7051126224695021,0.6404615502738565,0.5556101682897793,0.6394 (0.8858)
+0.8606666666666667,0.6458333333333334,0.9143749999999999,0.6581548557140771,0.8982611111111112,0.7062561199391592,0.6478927529204325,0.5643934390098356,0.6348 (0.8830)
+0.8610000000000001,0.6525000000000001,0.913125,0.6598441828821093,0.8984866666666667,0.7068181311417241,0.6524398798691053,0.5685641204592976,0.6821 (0.8778)
+0.8573333333333333,0.6541666666666668,0.908125,0.6484380143356185,0.8986122222222223,0.7069699730419063,0.6483267827506689,0.56122686790811,0.6749 (0.8789)
+0.8596666666666666,0.6583333333333333,0.9099999999999999,0.6571822160318406,0.8988322222222223,0.7073174907138248,0.6524590486254914,0.5688108387680031,0.6939 (0.8785)
+0.8606666666666667,0.6575,0.9114583333333334,0.6577423355364391,0.899246111111111,0.7081981075137119,0.6543504636689583,0.5697081734305843,0.6507 (0.8737)
+0.8605,0.6566666666666666,0.9114583333333333,0.6564514408304779,0.8996188888888887,0.70922834136052,0.6538226067613717,0.5686590284251849,0.6616 (0.8737)
+0.8596666666666666,0.6525000000000001,0.9114583333333334,0.6563432848008147,0.8997149999999999,0.7092200668896885,0.6517300560217784,0.5659829641747792,0.6645 (0.8720)
+0.86,0.6541666666666666,0.9114583333333333,0.6545771001339781,0.8997666666666667,0.7092727733232722,0.6526332818306073,0.5663481887380297,0.6540 (0.8718)
+0.8594999999999999,0.6533333333333333,0.9110416666666666,0.6539453725256957,0.8997455555555554,0.7095431783891384,0.6514651214789152,0.5652102208643273,0.6574 (0.8725)
+0.8604999999999999,0.6558333333333333,0.9116666666666667,0.6558553387963435,0.8997505555555556,0.7094970099969254,0.6539351264189786,0.5680973584252671,0.6550 (0.8725)

checkpoint-best.pth

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

config.json

@@ -0,0 +1 @@
+{"model_type": "ViT", "architecture": "vit_large_patch16", "input_size": 256, "num_classes": 5, "drop_path_rate": 0.2, "layer_decay": 0.65, "weight_decay": 0.05, "base_learning_rate": 0.005, "batch_size": 10, "epochs": 50, "remove_background": true}

dataset_info.json

@@ -0,0 +1 @@
+{"num_classes": 5, "classes": ["0-No DR", "1-Mild NPDR", "2-Moderate NPDR", "3-Severe NPDR", "4-PDR"], "num_training_images": 6999}

model_card.json

@@ -0,0 +1 @@
+{"model_name": "./models/eyepacs-2015-2019-12-11-2023-233729/-12-11-2023-233729", "model_type": "ViT", "description": "Fine-tuned vit_large_patch16 model for ./models/eyepacs-2015-2019-12-11-2023-233729/", "use_cases": ["RETfound validation", "Diabetic screening"], "limitations": ["for research/education use only"], "ethics": ["Ethics 1", "Ethics 2"], "training_data": "6999 images from 5 classes", "training_procedure": "Fine-tuned for 50 epochs with batch size 10 and base learning rate 0.005", "intended_use": "Intended for use with eyepacs-2015-2019 dataset", "authors": ["Author 1", "Author 2"], "references": ["Reference 1", "Reference 2"]}

requirements.txt

@@ -0,0 +1,27 @@
+torch==1.8.1+cu111
+timm==0.3.2
+torchvision==0.9.1+cu111
+torchaudio==0.8.1
+opencv-python>=4.5.3.56
+pandas>=0.25.3
+Pillow>=8.3.1
+protobuf>=3.17.3
+pycm>=3.2
+pydicom>=2.3.0
+scikit-image>=0.17.2
+scikit-learn>=0.24.2
+scipy>=1.5.4
+tensorboard>=2.6.0
+tensorboard-data-server>=0.6.1
+tensorboard-plugin-wit>=1.8.0
+tqdm>=4.62.1
+einops>=0.3.0
+h5py>=2.8.0
+imageio>=2.9.0
+matplotlib>=3.3.2
+tqdm>=4.51.0
+transformers>=3.5.1
+utils>=1.0.1
+Pygments>=2.7.4
+pytorch-msssim>=1.0.0
+toml

training_config.json

@@ -0,0 +1 @@
+{"batch_size": 10, "epochs": 50, "model": "vit_large_patch16", "base_learning_rate": 0.005, "layer_decay": 0.65, "weight_decay": 0.05, "drop_path_rate": 0.2, "input_size": 256, "num_classes": 5, "task": "./models/eyepacs-2015-2019-12-11-2023-233729/", "output_dir": "D:\\\\data\\\\archive\\\\eyepacs2015_2019\\\\dr-segmented-2000\\\\_datasets\\\\_dataset_12-11-2023-020118\\\\training_data\\outputs\\_artifacts_12-11-2023-233729", "world_size": 1, "finetune": "D:\\\\data\\\\RetFound\\\\weights\\\\RETFound_cfp_weights.pth", "rmbg": true}