# Double-click this file:
LAUNCH_GUI.bat
# Or run in terminal:
python tests\gui_test.pychmod +x LAUNCH_GUI.sh
./LAUNCH_GUI.sh- Click "Use Default" button (loads
models/final_model.pt) - Make sure dataset is set to
cifar10(or your trained dataset) - Click "Load Model"
- Wait for ✓ green checkmark
- Click "Browse" under Image Selection
- Choose any image file (JPG, PNG, etc.)
- Image preview appears automatically
- Click "🔍 Predict" button
- See results instantly:
- Main prediction in large green text
- Confidence percentage
- Top-5 predictions with visual bars
CIFAR-10 Classes:
✈️ airplane- 🚗 automobile
- 🐦 bird
- 🐱 cat
- 🦌 deer
- 🐕 dog
- 🐸 frog
- 🐴 horse
- 🚢 ship
- 🚛 truck
Your Results:
- Training Accuracy: 99.98%
- Validation Accuracy: 75.81%
- Model: ResNet18 (11.2M parameters)
- Professional dark background
- Green accent colors
- Smooth animations
- Easy-to-read fonts
- Loading indicators
- Progress bars
- Status messages
- Error handling
- Image preview
- Model information display
- Top-5 predictions with bars
- Confidence percentages
- Load Once, Test Many: Load model once, test unlimited images
- Quick Testing: Use default button for instant model loading
- Best Results: Use clear, centered images
- Fast Predictions: First prediction initializes, then super fast!
- Check Info: Model info shows parameters and accuracy
"No module named 'PyQt6'"
pip install PyQt6"Model file not found"
- Train a model first:
python train.py --dataset cifar10 --epochs 50 - Or check
models/folder exists
GUI won't start
pip install --upgrade PyQt6Prediction errors
- Ensure dataset name matches training dataset
- Check image file is valid
- Verify model loaded successfully (green checkmark)
For CIFAR-10:
- Photos of cats, dogs, horses
- Pictures of cars, trucks, airplanes
- Images of ships, frogs, birds
- Nature scenes with deer
Tips:
- Use clear, single-object images
- Centered subjects work best
- Good lighting improves accuracy
- Any image size works (auto-resized)
Based on your training:
- High confidence (>90%): Clear images of trained classes
- Medium confidence (50-90%): Partial views or similar classes
- Low confidence (<50%): Unclear or out-of-distribution images
- Test Different Images: Try various images from each class
- Check Accuracy: Compare predictions with actual labels
- Train More: Improve model with more epochs for better accuracy
- Try Other Datasets: Load models trained on different datasets
1. Start GUI
2. Click "Use Default"
3. Click "Load Model"
✓ Model loaded successfully
4. Click "Browse" → Select cat.jpg
5. Click "🔍 Predict"
Results:
🎯 cat
Confidence: 94.3%
Top-5:
1. cat ████████████████ 94.3%
2. dog ██ 3.2%
3. deer █ 1.5%
4. bird █ 0.7%
5. frog ░ 0.3%
Your model achieved 75.81% validation accuracy - test it on real images and see how it performs!
Questions or Issues?
- Check
tests/README_GUI.mdfor detailed documentation - Verify model file exists in
models/folder - Ensure PyQt6 is installed:
pip list | grep PyQt6
Made with 🔥 by NeuralForge