py3dcal 1.0.0__py3-none-any.whl

py3DCal/model_training/touchnet/touchnet_architecture.py

@@ -0,0 +1,72 @@
+import torch.nn as nn
+import torch.nn.functional as F
+
+class TouchNetArchitecture(nn.Module):
+    """
+    TouchNetArchitecture: A PyTorch neural network architecture for TouchNet.
+    """
+    def __init__(self):
+        super().__init__()
+
+        self.conv1 = nn.Conv2d(5, 32, kernel_size=7, padding=3)
+        self.bn1 = nn.BatchNorm2d(32)
+        self.dropout1 = nn.Dropout2d(0.2)
+
+        self.conv2 = nn.Conv2d(32, 64, kernel_size=7, padding=3)
+        self.bn2 = nn.BatchNorm2d(64)
+        self.dropout2 = nn.Dropout2d(0.2)
+
+        self.conv3 = nn.Conv2d(64, 128, kernel_size=7, padding=3)
+        self.bn3 = nn.BatchNorm2d(128)
+        self.dropout3 = nn.Dropout2d(0.2)
+
+        self.conv4 = nn.Conv2d(128, 256, kernel_size=5, padding=2)
+        self.bn4 = nn.BatchNorm2d(256)
+        self.dropout4 = nn.Dropout2d(0.3)
+
+        self.conv5 = nn.Conv2d(256, 256, kernel_size=5, padding=2)
+        self.bn5 = nn.BatchNorm2d(256)
+        self.dropout5 = nn.Dropout2d(0.3)
+
+        self.conv6 = nn.Conv2d(256, 128, kernel_size=5, padding=2)
+        self.bn6 = nn.BatchNorm2d(128)
+        self.dropout6 = nn.Dropout2d(0.2)
+
+        self.conv7 = nn.Conv2d(128, 64, kernel_size=3, padding=1)
+        self.bn7 = nn.BatchNorm2d(64)   
+        self.dropout7 = nn.Dropout2d(0.2)
+
+        self.conv8 = nn.Conv2d(64, 32, kernel_size=3, padding=1)
+        self.bn8 = nn.BatchNorm2d(32)
+        self.dropout8 = nn.Dropout2d(0.2)
+
+        self.conv9 = nn.Conv2d(32, 2, kernel_size=1)
+
+    def forward(self, x):
+        x = F.relu(self.bn1(self.conv1(x)))
+        x = self.dropout1(x)
+
+        x = F.relu(self.bn2(self.conv2(x)))
+        x = self.dropout2(x)
+
+        x = F.relu(self.bn3(self.conv3(x)))
+        x = self.dropout3(x)
+
+        x = F.relu(self.bn4(self.conv4(x)))
+        x = self.dropout4(x)
+
+        x = F.relu(self.bn5(self.conv5(x)))
+        x = self.dropout5(x)
+
+        x = F.relu(self.bn6(self.conv6(x)))
+        x = self.dropout6(x)
+
+        x = F.relu(self.bn7(self.conv7(x)))
+        x = self.dropout7(x)
+
+        x = F.relu(self.bn8(self.conv8(x)))
+        x = self.dropout8(x)
+
+        x = self.conv9(x)
+
+        return x

py3DCal/utils/utils.py

@@ -0,0 +1,32 @@
+import serial
+import sys
+import glob
+
+def list_com_ports():
+        """ Prints serial port names
+
+        :raises EnvironmentError:
+            On unsupported or unknown platforms
+        """
+        if sys.platform.startswith('win'):
+            ports = ['COM%s' % (i + 1) for i in range(256)]
+        elif sys.platform.startswith('linux') or sys.platform.startswith('cygwin'):
+            # this excludes your current terminal "/dev/tty"
+            ports = glob.glob('/dev/tty[A-Za-z]*')
+        elif sys.platform.startswith('darwin'):
+            ports = glob.glob('/dev/tty.*')
+        else:
+            raise EnvironmentError('Unsupported platform')
+
+        result = []
+        for port in ports:
+            try:
+                s = serial.Serial(port)
+                s.close()
+                result.append(port)
+            except (OSError, serial.SerialException):
+                pass
+        
+        print('Available COM Ports: ', end='')
+        print(result)
+        print('\n')

py3dcal-1.0.0.dist-info/LICENSE

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

py3dcal-1.0.0.dist-info/METADATA

@@ -0,0 +1,29 @@
+Metadata-Version: 2.1
+Name: py3dcal
+Version: 1.0.0
+Summary: UNKNOWN
+Home-page: https://github.com/rohankotanu/py3DCal
+Author: Rohan Kota
+Author-email: rohankota2026@u.northwestern.edu
+License: MIT
+Platform: UNKNOWN
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy>=1.0.0
+Requires-Dist: matplotlib>=3.0.0
+Requires-Dist: pandas>=2.0.0
+Requires-Dist: scipy>=1.0.0
+Requires-Dist: torch>=2.0.0
+Requires-Dist: torchvision>=0.23.0
+Requires-Dist: pyserial>=3.0
+Requires-Dist: opencv-python>=4.0.0
+Requires-Dist: pillow>=11.0.0
+Requires-Dist: tqdm>=4.0.0
+Requires-Dist: requests>=2.0.0
+Requires-Dist: scikit-learn>=1.0.0
+Requires-Dist: digit-interface>=0.2.1
+
+# For instructions on how to use this library, please visit https://rohankotanu.github.io/3DCal/
+

py3dcal-1.0.0.dist-info/RECORD

@@ -0,0 +1,44 @@
+py3DCal/__init__.py,sha256=a2eY2WoX_GOmAIlLTi-pmp3Xm4TVpNf_7H__FtQIDh0,697
+py3DCal/data_collection/Calibrator.py,sha256=pxj6gqrQLHTxbMsdUfsUxUgOgY8pLqKarzkYxjpxW58,11496
+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=i-2BQENm3tezpE0ErpY057pQfeH5F8iufWMbfrRx8_s,801
+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=1jr9nfpja_19gEflw_o0CGa9JC5Sz2PlhVZNzbBtMKM,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=wNjmUHr9m_44Yq-KPldjLdWrb6ecpwHnmXmEY8X1DyM,2986
+py3DCal/model_training/datasets/GelSightMini_dataset.py,sha256=EY-6yVj6rCmcIhNeui-ZCmsf14f1-GDXcGL4ZmTToGo,3039
+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=O7jEtArQukV-jssXLHEueRiII5hE01kv2OBn0HS82Dc,3246
+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/depthmaps.py,sha256=v-UL7ui8ZMJLMtwVqfrYIecS2KTGjRwPXa7mhFQLztk,2684
+py3DCal/model_training/lib/fast_poisson.py,sha256=Eh0Q_cr6r1MMppZ0xUMnQXpHTnZ-W6VQlwYcRREcN20,2198
+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_device.py,sha256=P_yFdAa0nj8camlEryol7AVtBos0gMhtalmAVLGwx08,778
+py3DCal/model_training/lib/validate_parameters.py,sha256=lOOw5zg4feQngVElr54Q0SHsF7bhb7-3MFF_iceZNfE,1368
+py3DCal/model_training/models/__init__.py,sha256=FTMTEyUglpRIFJ4hi3Z7y1KdScGDVVg_OxYXip4b8wg,42
+py3DCal/model_training/models/touchnet.py,sha256=JIfhwXX0uIk6UcN2tmCLoiH9il9U2IByjbjYIjQEIqc,8079
+py3DCal/model_training/touchnet/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+py3DCal/model_training/touchnet/dataset.py,sha256=L2n5mHa_90UeZSLluhDyS7yck5x2WfqaqRSouKFbBc4,2924
+py3DCal/model_training/touchnet/touchnet.py,sha256=c7FyDviwclaAQtak-QY5710r53aiczhvvUAfoizYw24,31236
+py3DCal/model_training/touchnet/touchnet_architecture.py,sha256=CGWodHCezCGVBPrjezj6574Sh0QQwSGWhRckUY8d7Hw,2137
+py3DCal/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+py3DCal/utils/utils.py,sha256=hgTyWZuBXfo9lxLnOLd0445Aw2-uARtKGXuBhZmz-Z0,995
+py3dcal-1.0.0.dist-info/LICENSE,sha256=D95ljbgz6PW9niwHP26EWFN77QBvepSCsMKGp0mRVFM,1066
+py3dcal-1.0.0.dist-info/METADATA,sha256=RGWNQiOoZ0vAwjd1jq3mBLJO6eWK0UOEAb2KyrMr9fU,882
+py3dcal-1.0.0.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+py3dcal-1.0.0.dist-info/entry_points.txt,sha256=_N1ruxvLEyZmSAaPsCx8kEzbYSJ5bHG5S8bvpua_X5E,59
+py3dcal-1.0.0.dist-info/top_level.txt,sha256=NbatjyXjN_E6UMifZpkx-ohahGQH_ZFvqovwmvU7FMA,8
+py3dcal-1.0.0.dist-info/RECORD,,

py3dcal-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: bdist_wheel (0.45.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

py3dcal-1.0.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+list-com-ports = py3DCal:list_com_ports
+

py3dcal-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+py3DCal