image-classification-tools 0.5.6__tar.gz

image_classification_tools-0.5.6/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 George Perdrizet
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

image_classification_tools-0.5.6/PKG-INFO

@@ -0,0 +1,187 @@
+Metadata-Version: 2.4
+Name: image-classification-tools
+Version: 0.5.6
+Summary: A lightweight PyTorch toolkit for building and training image classification models
+License: GPLv3
+License-File: LICENSE
+Keywords: Python,Machine learning,Deep learning,CNNs,Computer vision,Image classification,PyTorch,Neural networks
+Author: gperdrizet
+Author-email: george@perdrizet.org
+Requires-Python: >=3.10,<3.13
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Education
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: GNU General Public License v3 (GPLv3)
+Classifier: License :: Other/Proprietary License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Scientific/Engineering :: Image Recognition
+Provides-Extra: tensorflow
+Requires-Dist: numpy (>=1.24)
+Requires-Dist: torch (>=2.0)
+Requires-Dist: torchvision (>=0.15)
+Project-URL: Documentation, https://gperdrizet.github.io/CIFAR10
+Project-URL: Homepage, https://github.com/gperdrizet/CIFAR10
+Project-URL: Issues, https://github.com/gperdrizet/CIFAR10/issues
+Project-URL: PyPI, https://pypi.org/project/image-classification-tools
+Project-URL: Repository, https://github.com/gperdrizet/CIFAR10
+Description-Content-Type: text/markdown
+
+# Image Classification Tools
+
+A lightweight PyTorch toolkit for building and training image classification models.
+
+## Overview
+
+This package provides utilities for common image classification tasks:
+
+- **Data loading**: Flexible data loaders for torchvision datasets and custom image folders
+- **Model training**: Training loops with progress tracking and validation
+- **Evaluation**: Accuracy metrics, confusion matrices, and performance analysis
+- **Visualization**: Learning curves, probability distributions, and evaluation plots
+- **Hyperparameter optimization**: Optuna integration for automated model tuning
+
+## Installation
+
+```bash
+pip install image-classification-tools
+```
+
+## Quick start
+
+### Basic usage
+
+```python
+import torch
+from pathlib import Path
+from torchvision import datasets, transforms
+from image_classification_tools.pytorch.data import (
+    load_datasets, prepare_splits, create_dataloaders
+)
+from image_classification_tools.pytorch.training import train_model
+from image_classification_tools.pytorch.evaluation import evaluate_model
+
+# Define transforms
+transform = transforms.Compose([
+    transforms.ToTensor(),
+    transforms.Normalize((0.5,), (0.5,))
+])
+
+# Load datasets
+train_dataset, test_dataset = load_datasets(
+    data_source=datasets.MNIST,
+    train_transform=transform,
+    eval_transform=transform,
+    download=True,
+    root=Path('./data/mnist')
+)
+
+# Prepare splits
+train_dataset, val_dataset, test_dataset = prepare_splits(
+    train_dataset=train_dataset,
+    test_dataset=test_dataset,
+    train_val_split=0.8
+)
+
+# Create dataloaders
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+train_loader, val_loader, test_loader = create_dataloaders(
+    train_dataset, val_dataset, test_dataset,
+    batch_size=64,
+    preload_to_memory=True,
+    device=device
+)
+
+# Define model, criterion, optimizer
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
+model = torch.nn.Sequential(
+    torch.nn.Flatten(),
+    torch.nn.Linear(784, 128),
+    torch.nn.ReLU(),
+    torch.nn.Linear(128, 10)
+).to(device)
+
+criterion = torch.nn.CrossEntropyLoss()
+optimizer = torch.optim.Adam(model.parameters(), lr=1e-3)
+
+# Train with lazy loading (moves batches to device during training)
+history = train_model(
+    model=model,
+    train_loader=train_loader,
+    val_loader=val_loader,
+    criterion=criterion,
+    optimizer=optimizer,
+    device=device,
+    lazy_loading=True,  # Set False if data already on device
+    epochs=10
+)
+
+# Evaluate
+accuracy, predictions, labels = evaluate_model(model, test_loader)
+print(f'Test accuracy: {accuracy:.2f}%')
+```
+
+### Hyperparameter optimization
+
+```python
+from image_classification_tools.pytorch.hyperparameter_optimization import create_objective
+import optuna
+
+# Define search space
+search_space = {
+    'batch_size': [32, 64, 128],
+    'n_conv_blocks': (1, 3),
+    'initial_filters': [16, 32, 64],
+    'n_fc_layers': (1, 3),
+    'conv_dropout_rate': (0.1, 0.5),
+    'fc_dropout_rate': (0.3, 0.7),
+    'learning_rate': (1e-4, 1e-2, 'log'),
+    'optimizer': ['Adam', 'SGD'],
+    'weight_decay': (1e-6, 1e-3, 'log')
+}
+
+# Create objective function
+objective = create_objective(
+    data_dir='./data',
+    train_transform=transform,
+    eval_transform=transform,
+    n_epochs=20,
+    device=device,
+    num_classes=10,
+    in_channels=1,
+    search_space=search_space
+)
+
+# Run optimization
+study = optuna.create_study(direction='maximize')
+study.optimize(objective, n_trials=50)
+```
+
+## Requirements
+
+- Python ≥ 3.10
+- PyTorch ≥ 2.0.0
+- torchvision ≥ 0.15.0
+- numpy
+- matplotlib
+- optuna (optional, for hyperparameter optimization)
+
+## Documentation
+
+Full documentation is available at: https://gperdrizet.github.io/CIFAR10/
+
+## Demo project
+
+See a complete example of using this package for CIFAR-10 classification:
+https://github.com/gperdrizet/CIFAR10
+
+## License
+
+GPLv3
+

image_classification_tools-0.5.6/pyproject.toml

@@ -0,0 +1,55 @@
+[build-system]
+requires = ["poetry-core>=1.9.0"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.poetry]
+name = "image-classification-tools"
+version = "0.5.6"
+description = "A lightweight PyTorch toolkit for building and training image classification models"
+authors = ["gperdrizet <george@perdrizet.org>"]
+readme = "src/image_classification_tools/README.md"
+license = "GPLv3"
+packages = [{include = "image_classification_tools", from = "src"}]
+exclude = ["notebooks", "models", "logs"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "Intended Audience :: Education",
+    "Intended Audience :: Science/Research",
+    "License :: OSI Approved :: GNU General Public License v3 (GPLv3)",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Topic :: Scientific/Engineering :: Image Recognition",
+]
+keywords = ["Python", "Machine learning", "Deep learning", "CNNs", "Computer vision", "Image classification", "PyTorch", "Neural networks"]
+
+[tool.poetry.dependencies]
+python = ">=3.10,<3.13"
+torch = ">=2.0"
+torchvision = ">=0.15"
+numpy = ">=1.24"
+
+[tool.poetry.group.docs]
+optional = true
+
+[tool.poetry.group.docs.dependencies]
+sphinx = ">=7.0"
+sphinx-rtd-theme = ">=2.0"
+nbsphinx = ">=0.9"
+sphinx-autodoc-typehints = ">=1.25"
+ipykernel = ">=6.0"
+pandoc = ">=2.0"
+
+[tool.poetry.extras]
+tensorflow = ["tensorflow"]
+
+[tool.poetry.urls]
+Homepage = "https://github.com/gperdrizet/CIFAR10"
+Documentation = "https://gperdrizet.github.io/CIFAR10"
+Repository = "https://github.com/gperdrizet/CIFAR10"
+Issues = "https://github.com/gperdrizet/CIFAR10/issues"
+PyPI = "https://pypi.org/project/image-classification-tools"

image_classification_tools-0.5.6/src/image_classification_tools/README.md

@@ -0,0 +1,152 @@
+# Image Classification Tools
+
+A lightweight PyTorch toolkit for building and training image classification models.
+
+## Overview
+
+This package provides utilities for common image classification tasks:
+
+- **Data loading**: Flexible data loaders for torchvision datasets and custom image folders
+- **Model training**: Training loops with progress tracking and validation
+- **Evaluation**: Accuracy metrics, confusion matrices, and performance analysis
+- **Visualization**: Learning curves, probability distributions, and evaluation plots
+- **Hyperparameter optimization**: Optuna integration for automated model tuning
+
+## Installation
+
+```bash
+pip install image-classification-tools
+```
+
+## Quick start
+
+### Basic usage
+
+```python
+import torch
+from pathlib import Path
+from torchvision import datasets, transforms
+from image_classification_tools.pytorch.data import (
+    load_datasets, prepare_splits, create_dataloaders
+)
+from image_classification_tools.pytorch.training import train_model
+from image_classification_tools.pytorch.evaluation import evaluate_model
+
+# Define transforms
+transform = transforms.Compose([
+    transforms.ToTensor(),
+    transforms.Normalize((0.5,), (0.5,))
+])
+
+# Load datasets
+train_dataset, test_dataset = load_datasets(
+    data_source=datasets.MNIST,
+    train_transform=transform,
+    eval_transform=transform,
+    download=True,
+    root=Path('./data/mnist')
+)
+
+# Prepare splits
+train_dataset, val_dataset, test_dataset = prepare_splits(
+    train_dataset=train_dataset,
+    test_dataset=test_dataset,
+    train_val_split=0.8
+)
+
+# Create dataloaders
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+train_loader, val_loader, test_loader = create_dataloaders(
+    train_dataset, val_dataset, test_dataset,
+    batch_size=64,
+    preload_to_memory=True,
+    device=device
+)
+
+# Define model, criterion, optimizer
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
+model = torch.nn.Sequential(
+    torch.nn.Flatten(),
+    torch.nn.Linear(784, 128),
+    torch.nn.ReLU(),
+    torch.nn.Linear(128, 10)
+).to(device)
+
+criterion = torch.nn.CrossEntropyLoss()
+optimizer = torch.optim.Adam(model.parameters(), lr=1e-3)
+
+# Train with lazy loading (moves batches to device during training)
+history = train_model(
+    model=model,
+    train_loader=train_loader,
+    val_loader=val_loader,
+    criterion=criterion,
+    optimizer=optimizer,
+    device=device,
+    lazy_loading=True,  # Set False if data already on device
+    epochs=10
+)
+
+# Evaluate
+accuracy, predictions, labels = evaluate_model(model, test_loader)
+print(f'Test accuracy: {accuracy:.2f}%')
+```
+
+### Hyperparameter optimization
+
+```python
+from image_classification_tools.pytorch.hyperparameter_optimization import create_objective
+import optuna
+
+# Define search space
+search_space = {
+    'batch_size': [32, 64, 128],
+    'n_conv_blocks': (1, 3),
+    'initial_filters': [16, 32, 64],
+    'n_fc_layers': (1, 3),
+    'conv_dropout_rate': (0.1, 0.5),
+    'fc_dropout_rate': (0.3, 0.7),
+    'learning_rate': (1e-4, 1e-2, 'log'),
+    'optimizer': ['Adam', 'SGD'],
+    'weight_decay': (1e-6, 1e-3, 'log')
+}
+
+# Create objective function
+objective = create_objective(
+    data_dir='./data',
+    train_transform=transform,
+    eval_transform=transform,
+    n_epochs=20,
+    device=device,
+    num_classes=10,
+    in_channels=1,
+    search_space=search_space
+)
+
+# Run optimization
+study = optuna.create_study(direction='maximize')
+study.optimize(objective, n_trials=50)
+```
+
+## Requirements
+
+- Python ≥ 3.10
+- PyTorch ≥ 2.0.0
+- torchvision ≥ 0.15.0
+- numpy
+- matplotlib
+- optuna (optional, for hyperparameter optimization)
+
+## Documentation
+
+Full documentation is available at: https://gperdrizet.github.io/CIFAR10/
+
+## Demo project
+
+See a complete example of using this package for CIFAR-10 classification:
+https://github.com/gperdrizet/CIFAR10
+
+## License
+
+GPLv3

image_classification_tools-0.5.6/src/image_classification_tools/__init__.py

@@ -0,0 +1,6 @@
+"""Image Classification Tools - General-purpose PyTorch image classification utilities.
+
+This package provides tools for building, training, and evaluating image classification
+models with PyTorch. It includes utilities for data loading, model training, evaluation,
+hyperparameter optimization, and visualization.
+"""

image_classification_tools-0.5.6/src/image_classification_tools/pytorch/__init__.py

@@ -0,0 +1,46 @@
+'''PyTorch utilities for image classification.'''
+
+from image_classification_tools.pytorch.data import (
+    load_dataset,
+    prepare_splits,
+    create_dataloaders,
+    generate_augmented_data
+)
+from image_classification_tools.pytorch.evaluation import evaluate_model
+from image_classification_tools.pytorch.training import train_model
+from image_classification_tools.pytorch.plotting import (
+    plot_sample_images,
+    plot_learning_curves,
+    plot_confusion_matrix,
+    plot_class_probability_distributions,
+    plot_evaluation_curves,
+    plot_optimization_results
+)
+from image_classification_tools.pytorch.hyperparameter_optimization import (
+    create_cnn,
+    train_trial,
+    create_objective
+)
+
+__all__ = [
+    # Data loading and preprocessing
+    'load_dataset',
+    'prepare_splits',
+    'create_dataloaders',
+    'generate_augmented_data',
+    # Model evaluation
+    'evaluate_model',
+    # Model training
+    'train_model',
+    # Plotting and visualization
+    'plot_sample_images',
+    'plot_learning_curves',
+    'plot_confusion_matrix',
+    'plot_class_probability_distributions',
+    'plot_evaluation_curves',
+    'plot_optimization_results',
+    # Hyperparameter optimization
+    'create_cnn',
+    'train_trial',
+    'create_objective'
+]