Medical equipment recognition using deep transfer learning

Autor: Shi-Ting Wong, Chian-Wen Too, Wun-She Yap, Kok-Chin Khor
Rok vydání: 2022
Předmět:
Zdroj: Journal of Intelligent & Fuzzy Systems. 43:1001-1010
ISSN: 1875-8967
1064-1246
DOI: 10.3233/jifs-212786
Popis: With technological advancement, visual search has become an effective tool for searching important information by providing images. We propose a practical medical equipment recognition that can be used in visual search through deep transfer learning. We evaluated three deep learning models, i.e., VGG-16, ResNet-50, and Inception-v3, to recognise ten different classes of medical equipment. A data set consisting of 2,666 images had been collected and augmented to measure the models’ effectiveness. The models pre-trained with the ImageNet data set were transferred to the final models, and the last layers were replaced and trained with the collected data set. A grid search method was then used to find the best combination of hyperparameters, such as optimiser, batch size, epoch number, dropout rate, and learning rate. We tested the models using photos captured using smartphones. The results showed that Inception-v3 outperformed the other two models with the highest accuracy of 0.9454. This is the first study that uses deep transfer learning for recognising medical equipment to our best knowledge. Such recognition technology can potentially be implemented in visual search for helping consumers to check the validity of medical equipment.
Databáze: OpenAIRE
Externí odkaz:
Nepřihlášeným uživatelům se plný text nezobrazuje