- Python 3.11
- PyTorch 2.1.2
- OpenCV (cv2)
- Tensorboard
- NumPy
This project implements a Convolutional Neural Network (CNN) for image classification, specifically designed for classifying images related to Pesplanus.
- Ensure you have all the necessary requirements installed.
- Clone this repository to your local machine.
- Navigate to the project directory.
- Run the Python script responsible for training and testing the CNN.
- Monitor the training process using Tensorboard.
- Evaluate the model's performance on test data.
Pesplanus is a term used to describe a certain category of images. More information about the dataset and its characteristics can be found here.
- For training the CNN, use the provided training script (
train.py
). - Adjust hyperparameters and network architecture as needed.
- Monitor the training process using Tensorboard for insights into loss and accuracy trends.
- After training, evaluate the model on test data using the testing script (
test.py
). - Analyze performance metrics such as accuracy, precision, and recall.
Contributions are welcome! Please fork this repository and submit pull requests with any improvements or additional features.
This project is licensed under the MIT License - see the LICENSE file for details.