Project information
- Category: Deep Learning
- Team Project
- Technology Used: Pytorch python Convolutional Neural Network
- Date: October 2024 - December 2024
- Github: github/Imagine_classification_CNN
Project Overview
Trained a convolutional neural network (CNN) using PyTorch to classify endangered animal species with 82.87% accuracy. Preprocessed a dataset of 6,500 images and implemented data augmentation, dropout, and batch normalization to enhance model generalization and mitigate overfitting. Conducted iterative testing across five model architectures, optimizing hyperparameters and learning rates. Proposed real-world applications for wildlife conservation, including integration with CCTV and social media monitoring for illegal hunting detection.