Розробка нейромережевої системи детекції вогню на основі моделі MobileNetV2
Ключові слова:
MobileNetV2, комп’ютерний зір, класифікація зображень, вогонь, TensorFlowАнотація
У роботі реалізовано систему класифікації зображень для виявлення вогню з використанням глибокої згорткової нейронної мережі MobileNetV2. Створено власний генератор даних BalancedDataGenerator, що забезпечує балансування класів і поділ на тренувальну та валідаційну вибірки. Навчання виконано в два етапи: базове тренування верхніх шарів і подальше тонке донавчання останніх шарів базової моделі. Отримана модель досягла високої точності класифікації на валідаційних даних.
Посилання
MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications / A. Howard et al. URL: https://arxiv.org/abs/1704.04861 (дата звернення: 25.10.2025).
Chollet F. Deep Learning with Python, , Second Edition. Manning Publications, 2018. URL: https://books.google.com/books?id=XHpKEAAAQBAJ (дата звернення: 25.10.2025).
TensorFlow Documentation: MobileNetV2 and Transfer Learning Guides [Online]. URL: https://www.tensorflow.org/tutorials/images/transfer_learning (дата звернення: 25.10.2025).
