This repository contains two folders of Convolutional Neural Networks (CNNs) applied to different classification tasks.
Digit-Recognizer/
β
βββ ai_vs_real/
β βββ source.py
β
βββ digit_recogn/
β βββ source.py
- CNN model applied to pixel-based data (numerical vectors)
- Does not use
.jpgimages, but preprocessed pixel values - Similar to classic digit recognition problems (e.g. MNIST)
Features:
- Input: pixel vectors
- Model: simple CNN
- Task: digit classification
-
CNN model applied to real images (.jpg)
-
Classifies images into:
0β real art1β AI-generated art
Features:
- Input: RGB images
- Preprocessing: resizing, augmentations
- Model: multi-layer CNN
- Metric: F1-score
- Python
- PyTorch
- Pandas
- NumPy
- PIL
- Install dependencies:
pip install torch torchvision pandas pillow scikit-learn
- Run the desired script:
python source.py
This repository explores:
- the difference between numerical and image-based inputs
- how CNNs behave in different contexts
- a full pipeline: dataset β model β evaluation
This code is intended for learning and experimentation purposes and is not fully optimized for production use.