1. Classification Folder contains:
- Model folder
- "Train and Test.py" contains the code for Train , validation and Testing for the datasets provided.
- "CNN2d.h5" is saved model file size of 1.53Mb which is a Lightweight Architecture.
- "extracted_features_Using_DenseNet 121" contains the features we extracted from the original images using Densenet121 to train the model well.
- Output folder
- contains different Metric evaluations (ROC Curve , Confusion Matrix , Model loss and Accuracy plots , Accuracy, F1 score , Recall )
- Pictures folder
- 5 best images selected from testing dataset 1 and 2 separately , 10 best images and interpretability plot(CAM) from Validation showing its classification
- Data Profiling report
- the process of examining, analyzing, reviewing and summarizing data sets to gain insight into the quality of data
- Click the link above to view the Data Profiling Report in your browser Data Profiling Report
- For More Clear Understanding
- Click here to check the notebook Jupyter Notebook link
Note: Due to the fact that we employed programmatic code to save the data profiling report in HTML, please take note. Consequently, there are almost 10,000 lines of HTML code, making up 98% of the entire language. Just ignore it.
Classification Metrics
Metrics |
Value |
Accuracy |
98.85 |
Recall |
99.62 |
F1 Score |
98.87 |
Validation Dataset Showing Its Classification
Testing Dataset Showing Its Classification
Test Dataset 1
Test Dataset 2
2. Detection Folder contains:
- Weights folder
- contains the "best_ckpt.pt" yolov6 model saved.
- Output folder
- contains different Metric evaluations ( AP, MAP, IOU )
- Pictures folder
- 5 best images selected from testing dataset 1 and 2 separately , 10 best images from Validation showing its detection with confidence level
- CAM Plot folder
- contains 10 best images of interpretability plot(CAM) for Validation, test datasets
- dataset.yaml
- defining the configuration and structure of the custom dataset
- prediction_validation.csv
- showing bbox's with confidence level on validation dataset
- yolov6_detection.py
- contains the code for Train , Test and Validation on Dataset to perform Detection
Detection Metrics
Metrics |
Value |
Average Precision |
65.50 |
Mean Average Precision |
61.51 |
Intersection over Union |
0.50 |
Validation Dataset Showing Its Detection
Testing Dataset Showing Its Detection
Test Dataset 1
Test Dataset 2
Validation Dataset Showing Its Interpretability Plot (CAM)
Testing Dataset Showing Its Interpretability Plot (CAM)
Test Dataset 1
Test Dataset 2
3. Excel Sheet Submission:
- The Excel sheet containing the image IDs and predicted class labels of testing dataset 1 is saved as "test_dataset_1_predictions.xlsx"
click me to download "test_dataset_1_predictions.xlsx"