A desktop GUI tool to train and test image classifiers using TensorFlow and Tkinter. Created by CS42.org.
- Select folders of categorized images for training.
- Customize image size, number of epochs, and test split.
- Track training progress with animated accuracy plot.
- See final validation accuracy after training.
- Display 5 random prediction results from validation set.
- Save/load model in
.kerasformat. - Toggle between dark and light themes.
- Reset the program anytime.
├── airplane/
│ ├── 1.jpg
│ ├── 2.jpg
├── automobile/
│ ├── 1.jpg
│ ├── 2.jpg
pip install -r requirements.txtpython Image_Classifier_Code.py- 3 Conv2D + MaxPooling layers
- Dense layers with ReLU and Softmax
- Optimizer: Adam
- Loss: Sparse Categorical Crossentropy
- Trained model:
model_ui.keras - Class list:
model_ui_classes.txt
- Use commercially or personally
- Modify and redistribute
- Include this license
- Attribute the original author
No warranty is provided with this software.

