Glaucoma Segmentation

Diplomová práca: neurónové siete, ktoré vyznačia optický disk a jamku na snímkach oka.

Kroky spracovania od snímky oka po nájdený optický disk

O projekte

Glaukóm sa diagnostikuje aj podľa pomeru optickej jamky a optického disku. V diplomovej práci som navrhol, implementoval a vyhodnotil dve architektúry sietí na ich nájdenie: kaskádu, kde druhá sieť hľadá jamku vo vnútri disku nájdeného prvou, a sieť s dvoma dekodérmi a spoločným enkodérom.

Modely som trénoval na dátach ORIGA a testoval na DRISHTI-GS. Najlepší dosiahol Dice skóre 96,8 % pre disk a 89,7 % pre jamku.

Hlavné body

Kaskádová architektúra a architektúra s dvoma dekodérmi v PyTorch
Detekcia oblasti záujmu a trénovanie v polárnych súradniciach
Vysvetliteľnosť pomocou Grad-CAM

Ako to funguje

Prieskumná analýza dát ukázala nevyváženosť tried a veľmi rozdielnu kvalitu snímok, čo určilo predspracovanie
Model CenterNet najprv nájde oblasť okolo optického disku a segmentácia beží na tomto výreze v polárnych súradniciach
Vertikálny pomer jamky a disku z predikcií som porovnal s hodnotami nameranými odborníkmi
Mapy pravdepodobnosti, aktivačné mapy a Grad-CAM ukazujú, na čo sa siete pozerajú

Kedy: 2023 – 2024
Typ: Univerzitný
Vedenie: Vedúci práce doc. RNDr. Silvester Czanner, PhD.
Verzie: Python 3.10
PyTorch 2.0
Technológie: Python
PyTorch
OpenCV
NumPy
pandas
scikit-learn
Albumentations
Matplotlib
Jupyter
Weights & Biases

Galéria

Štyri snímky oka s obrysom od expertov, predikciou modelu a oboma obrysmi cez seba — Výsledky modelu s dvoma dekodérmi: vstup, referenčná maska, predikcia a skutočné (modré) a predikované (čierne) obrysy optického disku a jamky.

Diagram dvoch U-Net modelov za sebou: prvý vytvorí masku disku, ktorá oreže vstup pre druhý, a ten vytvorí masku jamky — Kaskádová architektúra: prvý U-Net segmentuje optický disk, jeho maska oreže obrázok a druhý U-Net v ňom nájde jamku. Dosiahla Dice skóre 96,8 % pre disk a 87,1 % pre jamku.

Diagram siete s jedným spoločným enkodérom a dvoma dekodérmi, pre disk a pre jamku — Architektúra s dvoma dekodérmi: jeden spoločný enkodér, dekodér pre optický disk a dekodér pre jamku. Dosiahla Dice skóre 96,8 % pre disk a 89,7 % pre jamku.

Čiarové grafy straty a Dice skóre počas epoch trénovania — Priebeh trénovania modelu s dvoma dekodérmi: strata, Dice skóre a ďalšie metriky na trénovacích a validačných dátach.

Grafy porovnávajúce predikovaný a expertmi nameraný pomer jamky a disku — Vertikálny pomer jamky a disku, hlavný ukazovateľ glaukómu, vypočítaný z predikcií a porovnaný s hodnotami od expertov.