CoolFace
Apppublic

Jasonschwartzman/children-drawings-classifier-daniel-fraidin

sourceHugging Facemitupdated 1y agoView on Hugging Face
0likes
App README

๐ŸŽจ Children's Drawings AI Classification System

Advanced AI system for analyzing children's drawings to identify potential psychological indicators using multiple state-of-the-art deep learning models.

๐ŸŽฏ Features

  • โ€”Binary Classification: Normal vs Problematic detection (up to 87% accuracy)
  • โ€”Multi-label Classification: Identifies 4 specific psychological indicators
  • โ€”Multiple Models: VGG19 and ConvNeXtV2 architectures
  • โ€”Comprehensive Analysis: Ensemble predictions from multiple models
  • โ€”Professional Reporting: Detailed analysis with recommendations

๐Ÿค– Models

Binary Classification Models

  • โ€”VGG19 Binary: 87.18% accuracy - Best for general screening
  • โ€”ConvNeXtV2 Binary: 85% accuracy - Lightweight and efficient

Multi-label Classification Models

  • โ€”VGG19 Multi-label: Traditional approach for indicator analysis
  • โ€”ConvNeXtV2 Multi-label: 67% avg F1-score - Superior detailed analysis

๐Ÿท๏ธ Psychological Indicators Analyzed

  1. 1.Problematic limbs: Issues with arm/leg representation and positioning
  2. 2.Proportions: Disproportionate body parts indicating developmental concerns
  3. 3.Missing body parts: Omitted essential body elements (head, limbs, torso)
  4. 4.Distorted human figure: Overall figure distortion and unusual representations

๐Ÿ“Š Usage

  1. 1.Upload Image: Select a children's drawing (PNG, JPG, JPEG)
  2. 2.Choose Analysis: Complete analysis, Binary only, or Multi-label only
  3. 3.Select Model: ConvNeXtV2 (Modern) or VGG19 (Proven)
  4. 4.Get Results: Comprehensive AI analysis with professional insights

โš ๏ธ Important Disclaimers

  • โ€”Research Use Only: This tool is designed for research and educational purposes
  • โ€”Not Diagnostic: Should never be used as sole basis for psychological diagnosis
  • โ€”Professional Consultation: Always involve qualified mental health professionals
  • โ€”Ethical Use: Ensure proper consent and ethical guidelines when analyzing children's drawings

๐Ÿ”ฌ Technical Details

  • โ€”Training Data: 908 binary samples, 461 multilabel samples (augmented to 1816+)
  • โ€”Validation: Rigorous train/validation/test splits with stratification
  • โ€”Data Augmentation: Comprehensive augmentation (rotation, zoom, brightness, flipping)
  • โ€”Performance: Cross-validated training with early stopping and model checkpointing

๐Ÿ“– Citation

If you use this system in research, please cite appropriately and acknowledge the AI-assisted analysis nature.


Developed with advanced deep learning techniques for psychological research applications