rgai-air-pollution-image-classification

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

RGAI Air Pollution Image Classification by Aswin Candra

Example Usage

First of all, clone this repo.

from aswin_air_pollution import CustomCNN

model = CustomCNN.from_pretrained('aswincandra/rgai-air-pollution-image-classification')

Notebook example: here

Model description

This model is trying to reproduce the architecture of Utomo Sapdo et al. on the Air Pollution Image Dataset from India and Nepal Kaggle dataset. It achieves the following results on the testing set:

• Loss: 0.5276
• Accuracy: 0.8166

Architecture

Quoted from Utomo Sapdo et al.

The proposed model accepts (224 x 224 x 3) RGB images as inputs. The initial model block is comprised of two CNN layers with 64 filters and one maxpooling layer. The second model block contains two CNN layers with 128 filters and one maxpooling layer. The third through fifth blocks use modified residual blocks that only apply one CNN layer, with the output of that CNN layer being added to the previous maxpooling output before being transmitted to the maxpooling layer. All maxpooling layers employ a kernel size of 3x3. Instead of ReLU, the activation function for all CNN layers is LeakyReLU. These five blocks are utilized for image feature extraction. The extracted features will then be flattened and transmitted to FC layer sets. The first FC layer consists of 256 neurons, whereas the second FC layer consists of 128 neurons. Additionally, these two FC layers use LeakyReLU as an activation function.

Then the output from the last layer that described above fed to final Fully-connected Layer with 6 outputs, depicts the number of class labels.

Output labels dictionary

• '0': 'a_Good'
• '1': 'b_Moderate'
• '2': 'c_Unhealthy_for_Sensitive_Groups'
• '3': 'd_Unhealthy'
• '4': 'e_Very_Unhealthy'
• '5': 'f_Severe'

Training hyperparameters

The following hyperparameters were used during training:

• learning_rate: 1e-4
• train_batch_size: 16
• eval_batch_size: 16
• optimizer: Adam
• num_epochs: 15

Training results