This repository demonstrates a simple CNN implemented in PyTorch for binary classification of smiles in celebrity face images from the CelebA dataset. This notebook covers data loading, preprocessing with transformations and augmentation, model building, training, and evaluation. The goal; predict wether the person is smiling based on facial attr.
machine-learning computer-vision image-processing pytorch neural-networks facial-recognition convolutional-neural-networks celeba celeba-dataset cnn-classification face-attributes
-
Updated
Oct 2, 2025 - Jupyter Notebook