Detecting artifacts in endoscopy images using YOLOv2 — built during a summer internship based on the EAD2019 challenge.
Detects 7 types of artifacts in endoscopy frames: specularity, saturation, artifact, contrast, blur, bubbles, and instruments. Uses YOLOv2 via the Darknet framework for real-time object detection.
Compared multiple architectures (YOLO, Fast R-CNN, R-CNN). YOLOv2 had the best IoU and mAP scores, consistent with this paper.
# Follow the Jupyter notebook
endoscopy_artifact_detection.ipynbDownload the dataset from Google Drive (2,147 annotated frames).
Python, YOLOv2, Darknet (C), OpenCV, Jupyter