A Lightweight Multi-label Image Classification Model Based on Inception Module
| Autor: | Kusum Kumari Bharti, Aparajita Ojha, Poornima S. Thakur, Pritee Khanna, Shreya Jain |
|---|---|
| Rok vydání: | 2021 |
| Předmět: |
Contextual image classification
Computer science business.industry Deep learning 0211 other engineering and technologies 02 engineering and technology computer.software_genre Convolutional neural network Convolution Image (mathematics) ComputingMethodologies_PATTERNRECOGNITION 0202 electrical engineering electronic engineering information engineering Benchmark (computing) 020201 artificial intelligence & image processing Artificial intelligence Data mining business computer 021101 geological & geomatics engineering |
| Zdroj: | Communications in Computer and Information Science ISBN: 9789811611025 CVIP (3) |
| DOI: | 10.1007/978-981-16-1103-2_20 |
| Popis: | Convolutional Neural Networks (CNNs) have shown enormous potential for solving multi-label image classification problems. In recent years, a lot of experimentation is done with various state-of-the-art CNN architectures. The CNN architectures have evolved to become deeper and more complex in these years. These architectures are big due to a greater number of layers and trainable parameters. However, there are many real-time applications which demand fast and accurate classification. Keeping this in consideration, a simple model inspired by Inception V7 is proposed for multi-label image classification in this work. The proposed model consists of six convolution layers including three inception blocks with one million parameters approximately, which are very few as compared to many state-of-the-art CNN models. This makes the model deployable in lightweight devices for some real-time applications. The comparison experiments with other deep state-of-the-art CNNs were carried out on image datasets from multiple domains including general benchmark datasets, medical datasets, and agricultural datasets. The model exhibits better performance on many datasets making it feasible to use in various domains for multi-label image classification. |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|