Skip to content

About

This project trains a Convolutional Neural Network (CNN) using PyTorch on a Roboflow image dataset for multi-class image classification. It includes dataset loading, model training, evaluation, and prediction on new images, making it a complete deep learning pipeline.

Resources

Stars

0 stars

Watchers

1 watching

Forks

Latest commit

 

History

3 Commits

Folders and files

Repository files navigation

CNN Image Classification with PyTorch

📌 Project Overview

This project trains a Convolutional Neural Network (CNN) using PyTorch on a Roboflow image dataset for image classification tasks. My current project classifies ultrasound images of the tongue into speech sounds

🔧 Setup

1️⃣ Create and Activate Virtual Environment

python -m venv myenv
source myenv/bin/activate  # Mac/Linux
myenv\Scripts\activate     # Windows

**### 2️⃣ Install Dependencies**
pip install torch torchvision torchaudio
pip install numpy matplotlib pillow opencv-python
pip install tqdm scikit-learn roboflow

About

This project trains a Convolutional Neural Network (CNN) using PyTorch on a Roboflow image dataset for multi-class image classification. It includes dataset loading, model training, evaluation, and prediction on new images, making it a complete deep learning pipeline.

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages