py3dcal 1.0.0__py3-none-any.whl → 1.0.6__py3-none-any.whl

py3DCal/model_training/lib/validate_parameters.py

@@ -1,3 +1,4 @@
+import os
 import torch
 from typing import Union
 from pathlib import Path
@@ -23,7 +24,7 @@ def validate_device(device: str):
                 "See: https://pytorch.org/docs/stable/tensor_attributes.html#torch.device"
             ) from e
         
-def validate_root(root):
+def validate_root(root, must_exist=False):
     """
     Validates the root path specified by the user.
 
@@ -42,4 +43,45 @@ def validate_root(root):
     if not isinstance(root, (str, Path)):
        raise ValueError(
            "root directory must be a valid file system path as a string or pathlib.Path object\n"
-       )
+       )
+    
+    if must_exist and not os.path.exists(root):
+         raise ValueError(
+              f"root directory '{root}' does not exist.\n"
+         )
+
+def validate_dataset(root, subtract_blank: bool):
+    """
+    Validates the dataset path specified by the user.
+
+    Args:
+        root: Dataset path specified by the user.
+        subtract_blank (bool): Whether to subtract blank image.
+    Returns:
+        None.
+    Raises:
+        FileNotFoundError: If necessary files do not exist.
+    """
+    validate_root(root)
+
+    annotation_path = os.path.join(root, "annotations", "annotations.csv")
+    metadata_path = os.path.join(root, "annotations", "metadata.json")
+    probe_images_path = os.path.join(root, "probe_images")
+    blank_image_path = os.path.join(root, "blank_images", "blank.png")
+
+    # Check if root directory exists
+    if not os.path.exists(root):
+        raise FileNotFoundError(f"Dataset root directory '{root}' does not exist.")
+
+    # Check if all the necessary files exist
+    if not os.path.exists(annotation_path):
+        raise FileNotFoundError(f"annotations.csv file not found in annotations/ directory. Use py3DCal.annotations() function to create it.")
+
+    if not os.path.exists(metadata_path):
+        raise FileNotFoundError(f"metadata.json file not found in annotations/ directory. Use py3DCal.annotations() function to create it.")
+
+    if not os.path.exists(probe_images_path):
+        raise FileNotFoundError(f"probe_images/ directory not found in dataset root.")
+
+    if subtract_blank and not os.path.exists(blank_image_path):
+        raise FileNotFoundError(f"blank.png file not found in blank_images/ directory.")

{py3dcal-1.0.0.dist-info → py3dcal-1.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: py3dcal
-Version: 1.0.0
+Version: 1.0.6
 Summary: UNKNOWN
 Home-page: https://github.com/rohankotanu/py3DCal
 Author: Rohan Kota

py3dcal-1.0.6.dist-info/RECORD

@@ -0,0 +1,40 @@
+py3DCal/__init__.py,sha256=roGUqVPYu3kB2TMhvWqGo95GyyJxH-zhlWgnJQF4TvM,836
+py3DCal/data_collection/Calibrator.py,sha256=X9LaLrIPv2maBZR3CwZuWrtKISjsP3GxNEx8UvwVbmc,11586
+py3DCal/data_collection/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+py3DCal/data_collection/printers/Printer.py,sha256=ouqgWuJWk8PPjhTRFwolnupXbE0SzO819LIgw1ug-7s,1628
+py3DCal/data_collection/printers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+py3DCal/data_collection/printers/Ender3/Ender3.py,sha256=pkE9Kt-mMH-fpZC5Gl6YyDPAxOdfZxp4Z7s68N7D_is,2239
+py3DCal/data_collection/printers/Ender3/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+py3DCal/data_collection/sensors/Sensor.py,sha256=o4tdkPMU2RF27OcOoxLABE4-K7R7Cf-d7kiT-KEDwAk,1063
+py3DCal/data_collection/sensors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+py3DCal/data_collection/sensors/DIGIT/DIGIT.py,sha256=1Jc1qXwinsOFjJ6-QUoUcXOBFjG4M6ylnUjbWMe3NSc,1268
+py3DCal/data_collection/sensors/DIGIT/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+py3DCal/data_collection/sensors/DIGIT/default.csv,sha256=OqjTE9b5uAEAxrHyjsxS7UyUWHgjO8yWPN0gVbBj7_Q,26728
+py3DCal/data_collection/sensors/GelsightMini/GelsightMini.py,sha256=nQ8JbiQypLCdTLM4WBNmXB_FFQ_yyxJ9gaV-OoE08Gg,1193
+py3DCal/data_collection/sensors/GelsightMini/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+py3DCal/data_collection/sensors/GelsightMini/default.csv,sha256=lavPHcJ6o4VkvMvOk7lcdRCp9dOJxg_VrPNayf9zVvM,26449
+py3DCal/model_training/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+py3DCal/model_training/datasets/DIGIT_dataset.py,sha256=IYRqWDawbTTe4IOVjZuKVr0yVNBe1XLGC6PoDxsTMfo,3017
+py3DCal/model_training/datasets/GelSightMini_dataset.py,sha256=H8Fr_4f3HDHLLl6KshRfqt0FP8-3d4n9XRK0xfPcH0k,3070
+py3DCal/model_training/datasets/__init__.py,sha256=vqrB177ZXrBmqDnL472EWleJS6Y-BxYEy2Ao9hWWDHc,137
+py3DCal/model_training/datasets/split_dataset.py,sha256=AzNJlTgcXGa9AdHJnVJYNEyv__OuNHZAMB76Haqc-io,1351
+py3DCal/model_training/datasets/tactile_sensor_dataset.py,sha256=LuSQYNHY5vkOihlJpG1PIs9EtxYtVW5x3ac6wgRTXLQ,3314
+py3DCal/model_training/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+py3DCal/model_training/lib/add_coordinate_embeddings.py,sha256=8wit43RIx28IKEB82SnH_Of09FAmiWM3jgOeXpamM1I,1198
+py3DCal/model_training/lib/annotate_dataset.py,sha256=J_CD4sc0vVp9Lq8cH-GWeoI98PwgU7dEV3bRLVAsFbA,21547
+py3DCal/model_training/lib/depthmaps.py,sha256=eW9rs4Ch64SwgwoTc1riauolX-IDxrYXmOJ0qQqQAgE,3183
+py3DCal/model_training/lib/fast_poisson.py,sha256=wJ5MTkSCxkFU3wUx-zomvIYPcAyEpPZj-LX7JQOx8JE,2252
+py3DCal/model_training/lib/get_gradient_map.py,sha256=IbCigrK_-6ZkeOSaHZAIhMu2pFmkSpWAaz1EjUtenCM,1438
+py3DCal/model_training/lib/precompute_gradients.py,sha256=zc1uvishZP7PjBWYF2VSrIMCtEkLrTPtLktOTpCh9P8,1860
+py3DCal/model_training/lib/train_model.py,sha256=fxFIfKWp3WA1Aa2IEczKBJCivVyVovj7IW2HqNw5IlE,4016
+py3DCal/model_training/lib/validate_parameters.py,sha256=My8cbGPSWsVpMB54jydbLYgfefWPMLGvr4oiPyLImb0,3052
+py3DCal/model_training/models/__init__.py,sha256=FTMTEyUglpRIFJ4hi3Z7y1KdScGDVVg_OxYXip4b8wg,42
+py3DCal/model_training/models/touchnet.py,sha256=JIfhwXX0uIk6UcN2tmCLoiH9il9U2IByjbjYIjQEIqc,8079
+py3DCal/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+py3DCal/utils/utils.py,sha256=hgTyWZuBXfo9lxLnOLd0445Aw2-uARtKGXuBhZmz-Z0,995
+py3dcal-1.0.6.dist-info/LICENSE,sha256=D95ljbgz6PW9niwHP26EWFN77QBvepSCsMKGp0mRVFM,1066
+py3dcal-1.0.6.dist-info/METADATA,sha256=xwSfMW_oF1trG9HODqx8aokzXveNxek9vN_IuP5JbQM,882
+py3dcal-1.0.6.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+py3dcal-1.0.6.dist-info/entry_points.txt,sha256=_N1ruxvLEyZmSAaPsCx8kEzbYSJ5bHG5S8bvpua_X5E,59
+py3dcal-1.0.6.dist-info/top_level.txt,sha256=NbatjyXjN_E6UMifZpkx-ohahGQH_ZFvqovwmvU7FMA,8
+py3dcal-1.0.6.dist-info/RECORD,,

py3DCal/model_training/lib/validate_device.py

@@ -1,22 +0,0 @@
-import torch
-
-def validate_device(device: str):
-        """
-        Validates the device by converting it to a torch.device object.
-        Args:
-            device (str): Device to run the model on.
-        Returns:
-            None.
-        Raises:
-            ValueError: If the device is not specified or invalid.
-        """
-        try:
-            device = torch.device(device)
-        except Exception as e:
-            raise ValueError(
-                f"Invalid device '{device}'. Valid options include:\n"
-                "  - 'cpu': CPU processing\n"
-                "  - 'cuda' or 'cuda:0': NVIDIA GPU\n"
-                "  - 'mps': Apple Silicon GPU\n"
-                "See: https://pytorch.org/docs/stable/tensor_attributes.html#torch.device"
-            ) from e

py3DCal/model_training/touchnet/dataset.py

@@ -1,78 +0,0 @@
-import os
-import sys
-import torch
-import numpy as np
-import pandas as pd
-from PIL import Image
-from torch.utils.data import Dataset
-from ..lib.precompute_gradients import precompute_gradients
-from ..lib.get_gradient_map import get_gradient_map
-
-
-class TactileSensorDataset(Dataset):
-    """
-    Tactile Sensor dataset.
-    """
-    def __init__(self, dataset_path, annotation_path, blank_image_path, transform=None, radius=36):
-        self.dataset_path = dataset_path
-        self.annotation_path = annotation_path
-        self.blank_image_path = blank_image_path
-        self.transform = transform
-        self.csv_file = dataset_path  # Store the CSV filename
-        
-        # Load the CSV data
-        self.data = pd.read_csv(annotation_path, comment='#')
-        
-        self.precomputed_gradients = precompute_gradients(dataset_path, annotation_path, r=radius)
-        self.blank_tensor = self.transform(Image.open(blank_image_path).convert("RGB"))
-    
-    def _add_coordinate_channels(self, image):
-        """
-        Add x and y coordinate channels to the input image.
-        X channel: column indices (1s in first column, 2s in second column, etc.)
-        Y channel: row indices (1s in first row, 2s in second row, etc.)
-        """
-        # Get image dimensions
-        _, height, width = image.shape
-        
-        # Create x coordinate channel (column indices)
-        x_coords = torch.arange(1, width + 1, dtype=torch.float32).unsqueeze(0).repeat(height, 1)
-        x_channel = x_coords.unsqueeze(0)  # Add channel dimension
-        
-        # Create y coordinate channel (row indices)
-        y_coords = torch.arange(1, height + 1, dtype=torch.float32).unsqueeze(1).repeat(1, width)
-        y_channel = y_coords.unsqueeze(0)  # Add channel dimension
-        
-        # Concatenate original image with coordinate channels
-        image_with_coords = torch.cat([image, x_channel, y_channel], dim=0)
-        
-        return image_with_coords
-        
-    def __len__(self):
-        return len(self.data)  # Use the DataFrame length
-
-    def __getitem__(self, idx):
-        # Check if index is valid
-        if idx < 0 or idx >= len(self.data):
-            raise IndexError("Index out of range")
-        
-        if torch.is_tensor(idx):
-            idx = idx.tolist()
-
-        img_name = os.path.join(self.dataset_path, self.data.iloc[idx, 0])
-        img = Image.open(img_name).convert("RGB")
-        image = np.array(img, copy=True)
-
-        target = get_gradient_map(idx, annotation_path=self.annotation_path, precomputed_gradients=self.precomputed_gradients)
-
-        # if self.transform:
-        image = self.transform(image)
-        target = self.transform(target)
-
-        # Subtract pre-transformed blank tensor
-        image = image - self.blank_tensor
-        # Add coordinate channels
-        image = self._add_coordinate_channels(image)
-
-        sample = (image, target)
-        return sample