homa 0.30000000000000004__tar.gz → 2.1__tar.gz

{homa-0.30000000000000004 → homa-2.1}/PKG-INFO

@@ -1,16 +1,15 @@
 Metadata-Version: 2.1
 Name: homa
-Version: 0.30000000000000004
+Version: 2.1
 Maintainer: Taha Shieenavaz
 Maintainer-email: tahashieenavaz@gmail.com
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: torchvision
+Requires-Dist: torch
 
-<p align="center">
-	<img
-		src="https://raw.githubusercontent.com/tahashieenavaz/homa/main/art/homa.svg"
-		width=400
-	/>
-</p>
+# Homa
 
-<hr />
+<div align="center">
+    <img src="art/homa.svg" width="500" />
+</div>

homa-2.1/README.md

@@ -0,0 +1,5 @@
+# Homa
+
+<div align="center">
+    <img src="art/homa.svg" width="500" />
+</div>

homa-2.1/homa/__init__.py

@@ -0,0 +1,2 @@
+from .datasets import ImageDataset
+from .datasets import AugmentedDataset

homa-2.1/homa/datasets.py

@@ -0,0 +1,71 @@
+import torch
+from torchvision import transforms
+
+
+class ImageDataset(torch.utils.data.Dataset):
+    def __init__(self, images, labels, transform=None):
+        self.images = images
+        self.labels = labels
+        self.transform = transform
+
+    def __len__(self):
+        return len(self.labels)
+
+    def __getitem__(self, idx):
+        image = self.images[idx]
+        label = self.labels[idx]
+
+        if self.transform:
+            image = self.transform(image)
+
+        return image, torch.tensor(label, dtype=torch.long)
+
+
+class AugmentedDataset(torch.utils.data.Dataset):
+    def __init__(self, dataset, transform_probability=0.6):
+        self.original_dataset = dataset
+        self.transform_probability = transform_probability
+        self.augmented_data = []
+        self.transformations = [
+            transforms.Compose(
+                [
+                    transforms.RandomHorizontalFlip(),
+                    transforms.ColorJitter(
+                        brightness=0.5, contrast=0.5, saturation=0.5, hue=0.2
+                    ),
+                    transforms.RandomRotation(45),
+                ]
+            ),
+            transforms.Compose(
+                [
+                    transforms.RandomVerticalFlip(),
+                    transforms.RandomAffine(
+                        degrees=30, translate=(0.1, 0.1), scale=(0.8, 1.2)
+                    ),
+                    transforms.GaussianBlur(kernel_size=(5, 5), sigma=(0.1, 2.0)),
+                ]
+            ),
+            transforms.Compose(
+                [
+                    transforms.RandomResizedCrop(size=(224, 224), scale=(0.5, 1.0)),
+                    transforms.RandomPerspective(distortion_scale=0.5, p=1.0),
+                    transforms.RandomGrayscale(p=0.3),
+                ]
+            ),
+        ]
+        self._augment_dataset()
+
+    def _augment_dataset(self):
+        for image, label in self.original_dataset:
+            if torch.rand(1).item() < self.transform_probability:
+                for transform in self.transformations:
+                    augmented_image = transform(image)
+                    self.augmented_data.append((augmented_image, label))
+            else:
+                self.augmented_data.append((image, label))
+
+    def __len__(self):
+        return len(self.augmented_data)
+
+    def __getitem__(self, idx):
+        return self.augmented_data[idx]

{homa-0.30000000000000004 → homa-2.1}/homa.egg-info/PKG-INFO

@@ -1,16 +1,15 @@
 Metadata-Version: 2.1
 Name: homa
-Version: 0.30000000000000004
+Version: 2.1
 Maintainer: Taha Shieenavaz
 Maintainer-email: tahashieenavaz@gmail.com
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: torchvision
+Requires-Dist: torch
 
-<p align="center">
-	<img
-		src="https://raw.githubusercontent.com/tahashieenavaz/homa/main/art/homa.svg"
-		width=400
-	/>
-</p>
+# Homa
 
-<hr />
+<div align="center">
+    <img src="art/homa.svg" width="500" />
+</div>

homa-2.1/homa.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+LICENSE
+README.md
+setup.py
+homa/__init__.py
+homa/datasets.py
+homa.egg-info/PKG-INFO
+homa.egg-info/SOURCES.txt
+homa.egg-info/dependency_links.txt
+homa.egg-info/requires.txt
+homa.egg-info/top_level.txt

homa-2.1/homa.egg-info/requires.txt

@@ -0,0 +1,2 @@
+torchvision
+torch

{homa-0.30000000000000004 → homa-2.1}/setup.py

@@ -9,7 +9,7 @@ with open("version.txt", "r") as fh:
     current_version = float(fh.readline())
 
 with open("version.txt", "w") as fh:
-    next_version = current_version + 0.1
+    next_version = round(current_version + 0.01, 2)
     fh.write(str(next_version))
 
 setup(
@@ -18,10 +18,7 @@ setup(
     maintainer_email="tahashieenavaz@gmail.com",
     version=next_version,
     packages=find_packages(),
-    install_requires=[
-        "opencv-python",
-        "numpy"
-    ],
+    install_requires=["torchvision", "torch"],
     long_description=description,
     long_description_content_type="text/markdown",
 )

homa-0.30000000000000004/README.md

@@ -1,8 +0,0 @@
-<p align="center">
-	<img
-		src="https://raw.githubusercontent.com/tahashieenavaz/homa/main/art/homa.svg"
-		width=400
-	/>
-</p>
-
-<hr />

homa-0.30000000000000004/homa/__init__.py

@@ -1 +0,0 @@
-from .main import *

homa-0.30000000000000004/homa/classes/Collection.py

@@ -1,11 +0,0 @@
-from typing import List
-
-
-class Collection:
-    def __init__(self, items: List[any]):
-        self.value = items
-
-    def map(self, callback: callable):
-        return list(
-            map(callback, self.value)
-        )

homa-0.30000000000000004/homa/classes/Logger.py

@@ -1,4 +0,0 @@
-class Logger:
-    @staticmethod
-    def danger(message: str) -> None:
-        print(message)

homa-0.30000000000000004/homa/classes/Repository.py

@@ -1,29 +0,0 @@
-from ..helpers import is_colab
-
-
-class RepositoryWrapper:
-    def __init__(self):
-        self.sigmaX = 0
-        self.sigmaY = 0
-
-        self.directory = "./"
-        self.images = {}
-        self.cameras = {}
-        self.window_counter = 0
-
-        if is_colab():
-            from google.colab.pathces import cv2_imshow as imshow
-        else:
-            from cv2 import imshow
-
-        self.imshow = imshow
-
-    def get_counter(self):
-        self.window_counter += 1
-        return self.window_counter
-
-    def __getattr__(self, name: str) -> any:
-        pass
-
-
-Repository = RepositoryWrapper()

homa-0.30000000000000004/homa/helpers.py

@@ -1,16 +0,0 @@
-import sys
-from typing import List
-from .classes.Collection import Collection
-from .classes.Logger import Logger
-
-
-def is_colab() -> bool:
-    return 'google.colab' in sys.modules
-
-
-def collection(items: List[any]):
-    return Collection(items)
-
-
-def danger(message: str) -> None:
-    Logger.danger(message)

homa-0.30000000000000004/homa/main.py

@@ -1,108 +0,0 @@
-import cv2
-import numpy
-from typing import List
-from .helpers import collection
-from .helpers import danger
-
-from .classes.Repository import Repository
-
-
-def path(directory: str) -> None:
-    Repository.directory = directory
-
-
-def write(key: str, filename: str) -> None:
-    cv2.imwrite(
-        filename=filename,
-        img=Repository.images[key]
-    )
-
-
-def save(key: str, filename: str) -> None:
-    write(key, filename)
-
-
-def image(filename: str, key: str | None = None, color: bool = True) -> None:
-    # TODO: add no extension in the file
-    if key is None:
-        key = filename
-
-    Repository.images[key] = cv2.imread(filename, int(color))
-    return Repository.images[key]
-
-
-def show(key: any = None) -> None:
-    # TODO: add functionality to distinguish between camera and images
-
-    if key is not None and not isinstance(key, str):
-        Repository.imshow(f"Window #{Repository.get_counter()}", key)
-
-    elif key is None:
-        for key, image in Repository.images.items():
-            Repository.imshow(key, image)
-
-    elif key is not None:
-        if key in Repository.images:
-            Repository.imshow(key, Repository.images[key])
-        else:
-            danger(f"No image found with key {key}")
-
-    cv2.waitKey(0)
-
-
-def camera():
-    capture = cv2.VideoCapture()
-    _, frame = capture.read()
-    Repository.camera_frame = frame
-
-
-def stack(keys: List[str], new_key: str, axis: int):
-    Repository.images[new_key] = numpy.concatenate(
-        collection(keys).map(lambda key: Repository.images[key]),
-        axis=axis
-    )
-
-
-def vstack(keys: List[str], new_key: str) -> None:
-    stack(keys, new_key, 1)
-
-
-def hstack(keys: List[str] | str, new_key: str | None = None):
-    if isinstance(keys, str) and new_key is None:
-        hstack([keys], keys)
-        return
-
-    stack(keys, new_key, 0)
-
-
-def blur(key: str, kernel: int | List[int] = (7, 7), new_key: str | None = None):
-    if new_key is None:
-        new_key = key
-
-    if isinstance(kernel, int):
-        kernel = (kernel, kernel)
-
-    Repository.images[new_key] = cv2.blur(
-        Repository.images[key],
-        kernel
-    )
-
-
-def sigma(x: float = 0, y: float = 0):
-    Repository.sigmaX = x
-    Repository.sigmaY = y
-
-
-def gaussian(key: str, kernel: None | List[int] = None, new_key: str | None = None):
-    if new_key is None:
-        new_key = key
-
-    if isinstance(kernel, int):
-        kernel = (kernel, kernel)
-
-    Repository.images[new_key] = cv2.GaussianBlur(
-        Repository.images[key],
-        kernel,
-        sigmaX=Repository.sigmaX,
-        sigmaY=Repository.sigmaY
-    )

homa-0.30000000000000004/homa.egg-info/SOURCES.txt

@@ -1,16 +0,0 @@
-LICENSE
-README.md
-setup.py
-homa/__init__.py
-homa/constants.py
-homa/helpers.py
-homa/main.py
-homa.egg-info/PKG-INFO
-homa.egg-info/SOURCES.txt
-homa.egg-info/dependency_links.txt
-homa.egg-info/requires.txt
-homa.egg-info/top_level.txt
-homa/classes/Collection.py
-homa/classes/Logger.py
-homa/classes/Repository.py
-homa/classes/__init__.py

homa-0.30000000000000004/homa.egg-info/requires.txt

@@ -1,2 +0,0 @@
-opencv-python
-numpy