BIRALAB
Báo cáo hội nghị2022

Detection and Classification Knee Osteoarthritis Algorithm using YOLOv3 and VGG-16 Models

Thuật toán phát hiện và phân loại thoái hóa khớp gối dùng mô hình YOLOv3 và VGG-16

Tác giả

Phat Nguyen Huu, Dat Nguyen Thanh, Thanh Le Thi Hai, Chu Duc Hoang, Hung Pham Viet, Cac Nguyen Trong

Tên thành viên BIRALAB được in đậm và liên kết tới trang cá nhân.

Tóm tắt

The paper proposes an algorithm to help doctors identify Knee Osteoarthritis automatically on X-ray images by deep learning method with the YOLOv3 model and VGG-16. We collect data including 2874 knee X-ray images taken from the OAI dataset which is a large data set of Knees and joints. In this paper, all data are labeled and tested by chiropractors at Bach Mai Hospital which will be preprocessed by applying the CLAHE algorithm to improve image quality. Next, the YOLOv3 modelis trained with a preprocessed dataset as input. It then predicts the location of the knee joint on the X-ray image automatically. Finally, we use VGG-16 to classify the image. Besides, we alsouse preprocessing methods to enhance the image quality before training, and thus the accuracy of the problem is also significantly improved. The results display that the proposed method achievesan accuracy of up to 89%