Klasifikacija bolesti temeljem rendgenskih snimki prsnog koša korištenjem metoda strojnog učenja

završni rad

• Naslov (engleski): Classification of diseases based on chest X-rays using machine learning methods
• Autor: Andrea Milanović
• Mentor: Alan Jović (mentor)
• Član povjerenstva: Alan Jović (predsjednik povjerenstva)
• Član povjerenstva: Hrvoje Nuić (član povjerenstva)
• Član povjerenstva: Eda Jovičić (član povjerenstva)
• Ustanova koja je dodijelila akademski / stručni stupanj: Sveučilište u Zagrebu; Fakultet elektrotehnike i računarstva; Zagreb
• Datum i država obrane: 2024-07-02, Hrvatska
• Znanstveno / umjetničko područje, polje i grana: TEHNIČKE ZNANOSTI; Računarstvo

Sažetak

Ovaj rad prikazuje primjenu konvolucijskih neuronskih mreža za klasifikaciju i detekciju
bolesti temeljem rendgenskih snimki prsnog koša. Prikazana je izgradnja dva
klasifikacijska te dva detekcijska modela i uspoređena je njihova uspješnost. Analizirane
su metrike točnosti, preciznosti i konfuzijske matrice za izgrađene modele. Za izradu
rada korišten je programski jezik Python te modeli ResNet, EfficientNet i YOLOv8.
Model ResNet postigao je točnost od 50 %. Model YOLO ukupno postiže preciznost od
20.5 % i odziv od 18.5 %, no za neka oboljenja poput kardiomegalije te vrijednosti prelaze
70 %.

Sažetak (engleski)

This paper presents the application of convolutional neural networks for the classification
and detection of diseases based on chest X-ray images. The construction of
two classification and two detection models is demonstrated, and their performance is
compared. Accuracy metrics, precision metrics and confusion matrices are analyzed for
the constructed models. Python programming language was used for the development
of the paper, along with ResNet, EfficientNet, and YOLOv8 models. The ResNet model
achieved an accuracy of 50 %. The YOLO model achieved an overall precision of 20.5 %
and a recall of 18.5 %, but for some diseases such as cardiomegaly, these values exceed
70 %.

• Ključne riječi: strojno učenje; konvolucijske neuronske mreže; klasifikacija slika; detekcija na slikama; rendgen prsnog koša
• Ključne riječi (engleski): machine learning; convolutional neural networks; image classification; image detection; chest X-ray
• Jezik: hrvatski
• URN:NBN: urn:nbn:hr:168:556377
• Studijski program: Naziv: Računarstvo; Vrsta studija: sveučilišni; Stupanj studija: preddiplomski; Akademski / stručni naziv: sveučilišni/a prvostupnik/ prvostupnica (baccalaureus/baccalaurea) inženjer/inženjerka računarstva (univ.bacc.ing.comp.)
• Vrsta resursa: Tekst
• Način izrade datoteke: Izvorno digitalna
• Prava pristupa: Zatvoreni pristup
• Datum i vrijeme pohrane: 2024-10-12 21:35:50