This project demonstrates how to create an image classification model using TensorFlow in a Jupyter Notebook to classify images based on emotions.
-
Setup and Import Libraries:
- Import TensorFlow, Keras, NumPy, and Matplotlib.
-
Load and Preprocess the Dataset:
- Load an emotion-labeled image dataset.
- Split the dataset into training and validation sets.
-
Build the Model:
- Create a Sequential model with Conv2D, MaxPooling2D, Flatten, Dense, and Dropout layers.
-
Compile the Model:
- Use the Adam optimizer and Sparse Categorical Crossentropy loss function.
-
Train the Model:
- Train the model using the training dataset and validate it with the validation dataset.