--- language: en tags: - image-classification - computer-vision - deep-learning - face-detection - inception datasets: - custom license: mit --- # Inception-style Face Classification Model 🎭 This model uses an Inception-style architecture to distinguish between real human faces and AI-generated faces. ## Model Description 📝 ### Model Architecture - Inception-style network with multi-scale feature processing - Input shape: (224, 224, 3) - Multiple inception modules with parallel pathways - Global average pooling - Dense layers with dropout for classification - Binary output with sigmoid activation ### Task Binary classification to determine if a face image is real (human) or AI-generated. ### Framework and Training - Framework: TensorFlow - Training Device: GPU - Training Dataset: Custom dataset of real and AI-generated faces - Validation Metrics: - Accuracy: 52.94% - Loss: 0.6913 ## Intended Use 🎯 ### Primary Intended Uses - Research in deepfake detection - Educational purposes in deep learning - Face authentication systems ### Out-of-Scope Uses - Production-level face verification - Legal or forensic applications - Stand-alone security systems ## Training Procedure 🔄 ### Training Details ```python optimizer = Adam(learning_rate=0.0001) loss = 'binary_crossentropy' metrics = ['accuracy'] ``` ### Training Hyperparameters - Learning rate: 0.0001 - Batch size: 32 - Dropout rate: 0.5 - Architecture: - Initial conv: 64 filters, 7x7 - Inception modules: [64, 128, 256, 512] filters - Dense layer: 256 units ## Evaluation Results 📊 ### Performance Metrics - Validation Accuracy: 52.94% - Validation Loss: 0.6913 ### Advantages Over ResNet Model - Better generalization - More stable learning process - Lower validation loss - Slightly higher accuracy ## Usage 💻 ```python from tensorflow.keras.models import load_model import cv2 import numpy as np # Load the model model = load_model('face_classification_model2') # Preprocess image def preprocess_image(image_path): img = cv2.imread(image_path) img = cv2.resize(img, (224, 224)) img = img / 255.0 return np.expand_dims(img, axis=0) # Make prediction image = preprocess_image('face_image.jpg') prediction = model.predict(image) is_real = prediction[0][0] > 0.5 ``` ## Ethical Considerations 🤝 This model is designed for research and educational purposes only. Users should: - Obtain proper consent when processing personal face images - Be aware of potential biases in face detection systems - Consider privacy implications when using face analysis tools - Not use this model for surveillance or harmful applications ## Technical Limitations ⚠️ 1. Current performance limitations: - Accuracy only slightly above random chance - May require ensemble methods for better results - Limited testing on diverse datasets 2. Recommended improvements: - Extended training with larger datasets - Implementation of data augmentation - Hyperparameter optimization - Transfer learning from pre-trained models ## Citation 📚 ```bibtex @software{face_classification_model2, author = {Your Name}, title = {Face Classification Model using Inception Architecture}, year = {2024}, publisher = {HuggingFace}, url = {https://huggingface.co/arsath-sm/face_classification_model2} } ``` ## Contributors 👥 - Arsath S.M - Faahith K.R.M - Arafath M.S.M University of Jaffna ## License 📄 This model is licensed under the MIT License.