christofy 0.0.1__tar.gz

christofy-0.0.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Aswin Christo
+
+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.

christofy-0.0.1/MANIFEST.in

@@ -0,0 +1,2 @@
+include README.md
+include LICENSE

christofy-0.0.1/PKG-INFO

@@ -0,0 +1,52 @@
+Metadata-Version: 2.4
+Name: christofy
+Version: 0.0.1
+Summary: A lightweight and modular PyTorch-based image classification package
+Author-email: Aswin Christo <aswinchristo17@gmail.com>
+License: MIT License
+        
+        Copyright (c) 2025 Aswin Christo
+        
+        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.
+        
+Project-URL: Homepage, https://github.com/aswinchristo/christofy
+Project-URL: Bug Tracker, https://github.com/aswinchristo/christofy/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Classifier: License :: OSI Approved :: MIT License
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Dynamic: license-file
+
+# Christofy 🧠📦
+
+**Christofy** is a lightweight and modular PyTorch-based image classification package designed for both training and prediction. Built on top of ResNet18, it supports both binary and multi-class classification with clean abstractions and an easy-to-use API.
+
+---
+
+## 🚀 Features
+
+- ✅ Train image classifiers using CNN (ResNet18)
+- ✅ Binary and multi-class classification support
+- ✅ Predict the class of a single image using saved model
+- ✅ Modular structure: clean separation between training and prediction
+- ✅ Ready to be installed as a Python package
+
+---

christofy-0.0.1/README.md

@@ -0,0 +1,15 @@
+# Christofy 🧠📦
+
+**Christofy** is a lightweight and modular PyTorch-based image classification package designed for both training and prediction. Built on top of ResNet18, it supports both binary and multi-class classification with clean abstractions and an easy-to-use API.
+
+---
+
+## 🚀 Features
+
+- ✅ Train image classifiers using CNN (ResNet18)
+- ✅ Binary and multi-class classification support
+- ✅ Predict the class of a single image using saved model
+- ✅ Modular structure: clean separation between training and prediction
+- ✅ Ready to be installed as a Python package
+
+---

christofy-0.0.1/pyproject.toml

@@ -0,0 +1,23 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "christofy"
+version = "0.0.1"
+authors = [
+    { name="Aswin Christo", email="aswinchristo17@gmail.com" }
+]
+description = "A lightweight and modular PyTorch-based image classification package"
+readme = "README.md"
+license = { file="LICENSE" }
+requires-python = ">=3.9"
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "Operating System :: OS Independent",
+    "License :: OSI Approved :: MIT License"
+]
+
+[project.urls]
+"Homepage" = "https://github.com/aswinchristo/christofy"
+"Bug Tracker" = "https://github.com/aswinchristo/christofy/issues"

christofy-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

christofy-0.0.1/src/christofy/__init__.py

@@ -0,0 +1,2 @@
+from .model_trainer import train_model
+from .predictor import predict_image

christofy-0.0.1/src/christofy/model_trainer.py

@@ -0,0 +1,122 @@
+import os
+import torch
+import torch.nn as nn
+import torch.optim as optim
+from torch.utils.data import DataLoader, random_split
+from torchvision import datasets, transforms, models
+import matplotlib.pyplot as plt
+import seaborn as sns
+from sklearn.metrics import confusion_matrix, classification_report
+from PIL import Image
+import numpy as np
+import json
+
+
+def train_cnn_classifier(data_dir, learning_rate=0.001, epochs=10, batch_size=32, output_dir="output"):
+    os.makedirs(output_dir, exist_ok=True)
+
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+    # Detect classes
+    classes = sorted(os.listdir(data_dir))
+    num_classes = len(classes)
+    print(f"Detected {num_classes} classes: {classes}")
+
+    # Transforms
+    transform = transforms.Compose([
+        transforms.Resize((224, 224)),
+        transforms.RandomHorizontalFlip(),
+        transforms.ToTensor(),
+        transforms.Normalize(mean=[0.485, 0.456, 0.406],
+                             std=[0.229, 0.224, 0.225])
+    ])
+
+    dataset = datasets.ImageFolder(root=data_dir, transform=transform)
+    class_to_idx = dataset.class_to_idx
+    with open(os.path.join(output_dir, 'class_map.json'), 'w') as f:
+        json.dump(class_to_idx, f)
+
+    train_size = int(0.8 * len(dataset))
+    val_size = len(dataset) - train_size
+    train_dataset, val_dataset = random_split(dataset, [train_size, val_size])
+
+    train_loader = DataLoader(train_dataset, batch_size=batch_size, shuffle=True)
+    val_loader = DataLoader(val_dataset, batch_size=batch_size, shuffle=False)
+
+    # Model
+    model = models.resnet18(pretrained=False)
+    model.fc = nn.Linear(model.fc.in_features, num_classes)
+    model = model.to(device)
+
+    # Loss and optimizer
+    criterion = nn.CrossEntropyLoss()
+    optimizer = optim.Adam(model.parameters(), lr=learning_rate)
+
+    train_losses, val_losses = [], []
+    all_preds, all_labels = [], []
+
+    for epoch in range(epochs):
+        model.train()
+        total_loss = 0
+        for images, labels in train_loader:
+            images, labels = images.to(device), labels.to(device)
+
+            outputs = model(images)
+            loss = criterion(outputs, labels)
+
+            optimizer.zero_grad()
+            loss.backward()
+            optimizer.step()
+
+            total_loss += loss.item()
+
+        train_losses.append(total_loss / len(train_loader))
+
+        model.eval()
+        val_loss = 0
+        correct, total = 0, 0
+        with torch.no_grad():
+            for images, labels in val_loader:
+                images, labels = images.to(device), labels.to(device)
+                outputs = model(images)
+                loss = criterion(outputs, labels)
+                val_loss += loss.item()
+
+                _, predicted = torch.max(outputs, 1)
+                correct += (predicted == labels).sum().item()
+                total += labels.size(0)
+
+                all_preds.extend(predicted.cpu().numpy())
+                all_labels.extend(labels.cpu().numpy())
+
+        val_losses.append(val_loss / len(val_loader))
+        acc = correct / total
+        print(f"Epoch [{epoch+1}/{epochs}], Loss: {train_losses[-1]:.4f}, Val Loss: {val_losses[-1]:.4f}, Acc: {acc:.4f}")
+
+    # Save model
+    torch.save(model.state_dict(), os.path.join(output_dir, "model.pth"))
+
+    # Plot losses
+    plt.figure()
+    plt.plot(train_losses, label="Train Loss")
+    plt.plot(val_losses, label="Val Loss")
+    plt.legend()
+    plt.title("Training and Validation Loss")
+    plt.savefig(os.path.join(output_dir, "loss_plot.png"))
+
+    # Confusion matrix
+    cm = confusion_matrix(all_labels, all_preds)
+    plt.figure(figsize=(6, 6))
+    sns.heatmap(cm, annot=True, fmt="d", xticklabels=classes, yticklabels=classes, cmap="Blues")
+    plt.xlabel("Predicted")
+    plt.ylabel("Actual")
+    plt.title("Confusion Matrix")
+    plt.savefig(os.path.join(output_dir, "confusion_matrix.png"))
+
+    # Classification report
+    report = classification_report(all_labels, all_preds, target_names=classes)
+    with open(os.path.join(output_dir, "classification_report.txt"), 'w') as f:
+        f.write(report)
+
+    print("Training complete. Outputs saved to:", output_dir)
+

christofy-0.0.1/src/christofy/predictor.py

@@ -0,0 +1,42 @@
+import os
+import torch
+import torch.nn as nn
+import torch.optim as optim
+from torch.utils.data import DataLoader, random_split
+from torchvision import datasets, transforms, models
+import matplotlib.pyplot as plt
+import seaborn as sns
+from sklearn.metrics import confusion_matrix, classification_report
+from PIL import Image
+import numpy as np
+import json
+
+def predict_class(model_path, image_path):
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+    with open(os.path.join(os.path.dirname(model_path), 'class_map.json'), 'r') as f:
+        class_to_idx = json.load(f)
+    idx_to_class = {v: k for k, v in class_to_idx.items()}
+
+    transform = transforms.Compose([
+        transforms.Resize((224, 224)),
+        transforms.ToTensor(),
+        transforms.Normalize(mean=[0.485, 0.456, 0.406],
+                             std=[0.229, 0.224, 0.225])
+    ])
+
+    image = Image.open(image_path).convert('RGB')
+    image = transform(image).unsqueeze(0).to(device)
+
+    model = models.resnet18(pretrained=False)
+    model.fc = nn.Linear(model.fc.in_features, len(idx_to_class))
+    model.load_state_dict(torch.load(model_path, map_location=device))
+    model = model.to(device)
+    model.eval()
+
+    with torch.no_grad():
+        outputs = model(image)
+        _, predicted = torch.max(outputs, 1)
+
+    predicted_class = idx_to_class[predicted.item()]
+    return predicted_class

christofy-0.0.1/src/christofy.egg-info/PKG-INFO

@@ -0,0 +1,52 @@
+Metadata-Version: 2.4
+Name: christofy
+Version: 0.0.1
+Summary: A lightweight and modular PyTorch-based image classification package
+Author-email: Aswin Christo <aswinchristo17@gmail.com>
+License: MIT License
+        
+        Copyright (c) 2025 Aswin Christo
+        
+        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.
+        
+Project-URL: Homepage, https://github.com/aswinchristo/christofy
+Project-URL: Bug Tracker, https://github.com/aswinchristo/christofy/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Classifier: License :: OSI Approved :: MIT License
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Dynamic: license-file
+
+# Christofy 🧠📦
+
+**Christofy** is a lightweight and modular PyTorch-based image classification package designed for both training and prediction. Built on top of ResNet18, it supports both binary and multi-class classification with clean abstractions and an easy-to-use API.
+
+---
+
+## 🚀 Features
+
+- ✅ Train image classifiers using CNN (ResNet18)
+- ✅ Binary and multi-class classification support
+- ✅ Predict the class of a single image using saved model
+- ✅ Modular structure: clean separation between training and prediction
+- ✅ Ready to be installed as a Python package
+
+---

christofy-0.0.1/src/christofy.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+LICENSE
+MANIFEST.in
+README.md
+pyproject.toml
+src/christofy/__init__.py
+src/christofy/model_trainer.py
+src/christofy/predictor.py
+src/christofy.egg-info/PKG-INFO
+src/christofy.egg-info/SOURCES.txt
+src/christofy.egg-info/dependency_links.txt
+src/christofy.egg-info/top_level.txt

christofy-0.0.1/src/christofy.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

christofy-0.0.1/src/christofy.egg-info/top_level.txt

@@ -0,0 +1 @@
+christofy