- In our Model we preparing dataset for Garbage classification , divided it into training and validation sets.
- We use ImageDataGenerator for data augmentation to enhance our model's ability to generalize.
- Then we try different model as base model (MobileNet , Xception) for feature extraction, and we add a global average pooling layer to reduce the spatial dimensions.
Features that we extracted from the pre-trained model are for both training and testing datasets. - Then we perform feature selection using a genetic algorithm.
- We initialize a population of binary encoded feature vectors, this feature vector representing the presence or absence of features.
- Our Fitness function is evaluated using a support vector machine (SVM) classifier on the selected features.
- we select parents based on their fitness. We randomly select two indices for each member of the population, compare their fitness, and select the one with higher fitness as a parent.
- After that we perform a single-point crossover method between two parents that we selected to create two children.
- We randomly select a crossover point and combine the genetic information of the two parents before and after that point.
- And then apply mutation, and finally generate a new populations iteratively.
- We initialize a population of binary encoded feature vectors, this feature vector representing the presence or absence of features.
- After Selecting best features that suit our data we apply SVM to get final results.
-
Notifications
You must be signed in to change notification settings - Fork 0
Aalaa4444/Garbage_Classification
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
CNN model to classify garbage
Topics
image
svm
genetic-algorithm
image-processing
feature-selection
feature-extraction
confusion-matrix
pretrained-models
cnn-model
svm-classifier
classification-report
mobilenet
xception-model
cnn-classification
precision-recall
mobilenet-model
mobilenet-v3
image-data-generator
genetic-algorithm-from-scratch
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published