py3dcal 1.0.3__py3-none-any.whl → 1.0.6__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- py3DCal/__init__.py +6 -5
- py3DCal/data_collection/Calibrator.py +10 -8
- py3DCal/data_collection/{Sensors → sensors}/GelsightMini/GelsightMini.py +5 -5
- py3DCal/data_collection/{Sensors → sensors}/Sensor.py +10 -1
- py3DCal/model_training/datasets/tactile_sensor_dataset.py +4 -3
- py3DCal/model_training/lib/annotate_dataset.py +560 -0
- py3DCal/model_training/lib/depthmaps.py +11 -3
- py3DCal/model_training/lib/validate_parameters.py +44 -2
- {py3dcal-1.0.3.dist-info → py3dcal-1.0.6.dist-info}/METADATA +1 -1
- py3dcal-1.0.6.dist-info/RECORD +40 -0
- py3DCal/model_training/lib/validate_device.py +0 -22
- py3DCal/model_training/touchnet/__init__.py +0 -0
- py3DCal/model_training/touchnet/dataset.py +0 -78
- py3DCal/model_training/touchnet/touchnet.py +0 -736
- py3DCal/model_training/touchnet/touchnet_architecture.py +0 -72
- py3dcal-1.0.3.dist-info/RECORD +0 -44
- /py3DCal/data_collection/{Printers → printers}/Ender3/Ender3.py +0 -0
- /py3DCal/data_collection/{Printers → printers}/Ender3/__init__.py +0 -0
- /py3DCal/data_collection/{Printers → printers}/Printer.py +0 -0
- /py3DCal/data_collection/{Printers → printers}/__init__.py +0 -0
- /py3DCal/data_collection/{Sensors → sensors}/DIGIT/DIGIT.py +0 -0
- /py3DCal/data_collection/{Sensors → sensors}/DIGIT/__init__.py +0 -0
- /py3DCal/data_collection/{Sensors → sensors}/DIGIT/default.csv +0 -0
- /py3DCal/data_collection/{Sensors → sensors}/GelsightMini/__init__.py +0 -0
- /py3DCal/data_collection/{Sensors → sensors}/GelsightMini/default.csv +0 -0
- /py3DCal/data_collection/{Sensors → sensors}/__init__.py +0 -0
- {py3dcal-1.0.3.dist-info → py3dcal-1.0.6.dist-info}/LICENSE +0 -0
- {py3dcal-1.0.3.dist-info → py3dcal-1.0.6.dist-info}/WHEEL +0 -0
- {py3dcal-1.0.3.dist-info → py3dcal-1.0.6.dist-info}/entry_points.txt +0 -0
- {py3dcal-1.0.3.dist-info → py3dcal-1.0.6.dist-info}/top_level.txt +0 -0
|
@@ -1,72 +0,0 @@
|
|
|
1
|
-
import torch.nn as nn
|
|
2
|
-
import torch.nn.functional as F
|
|
3
|
-
|
|
4
|
-
class TouchNetArchitecture(nn.Module):
|
|
5
|
-
"""
|
|
6
|
-
TouchNetArchitecture: A PyTorch neural network architecture for TouchNet.
|
|
7
|
-
"""
|
|
8
|
-
def __init__(self):
|
|
9
|
-
super().__init__()
|
|
10
|
-
|
|
11
|
-
self.conv1 = nn.Conv2d(5, 32, kernel_size=7, padding=3)
|
|
12
|
-
self.bn1 = nn.BatchNorm2d(32)
|
|
13
|
-
self.dropout1 = nn.Dropout2d(0.2)
|
|
14
|
-
|
|
15
|
-
self.conv2 = nn.Conv2d(32, 64, kernel_size=7, padding=3)
|
|
16
|
-
self.bn2 = nn.BatchNorm2d(64)
|
|
17
|
-
self.dropout2 = nn.Dropout2d(0.2)
|
|
18
|
-
|
|
19
|
-
self.conv3 = nn.Conv2d(64, 128, kernel_size=7, padding=3)
|
|
20
|
-
self.bn3 = nn.BatchNorm2d(128)
|
|
21
|
-
self.dropout3 = nn.Dropout2d(0.2)
|
|
22
|
-
|
|
23
|
-
self.conv4 = nn.Conv2d(128, 256, kernel_size=5, padding=2)
|
|
24
|
-
self.bn4 = nn.BatchNorm2d(256)
|
|
25
|
-
self.dropout4 = nn.Dropout2d(0.3)
|
|
26
|
-
|
|
27
|
-
self.conv5 = nn.Conv2d(256, 256, kernel_size=5, padding=2)
|
|
28
|
-
self.bn5 = nn.BatchNorm2d(256)
|
|
29
|
-
self.dropout5 = nn.Dropout2d(0.3)
|
|
30
|
-
|
|
31
|
-
self.conv6 = nn.Conv2d(256, 128, kernel_size=5, padding=2)
|
|
32
|
-
self.bn6 = nn.BatchNorm2d(128)
|
|
33
|
-
self.dropout6 = nn.Dropout2d(0.2)
|
|
34
|
-
|
|
35
|
-
self.conv7 = nn.Conv2d(128, 64, kernel_size=3, padding=1)
|
|
36
|
-
self.bn7 = nn.BatchNorm2d(64)
|
|
37
|
-
self.dropout7 = nn.Dropout2d(0.2)
|
|
38
|
-
|
|
39
|
-
self.conv8 = nn.Conv2d(64, 32, kernel_size=3, padding=1)
|
|
40
|
-
self.bn8 = nn.BatchNorm2d(32)
|
|
41
|
-
self.dropout8 = nn.Dropout2d(0.2)
|
|
42
|
-
|
|
43
|
-
self.conv9 = nn.Conv2d(32, 2, kernel_size=1)
|
|
44
|
-
|
|
45
|
-
def forward(self, x):
|
|
46
|
-
x = F.relu(self.bn1(self.conv1(x)))
|
|
47
|
-
x = self.dropout1(x)
|
|
48
|
-
|
|
49
|
-
x = F.relu(self.bn2(self.conv2(x)))
|
|
50
|
-
x = self.dropout2(x)
|
|
51
|
-
|
|
52
|
-
x = F.relu(self.bn3(self.conv3(x)))
|
|
53
|
-
x = self.dropout3(x)
|
|
54
|
-
|
|
55
|
-
x = F.relu(self.bn4(self.conv4(x)))
|
|
56
|
-
x = self.dropout4(x)
|
|
57
|
-
|
|
58
|
-
x = F.relu(self.bn5(self.conv5(x)))
|
|
59
|
-
x = self.dropout5(x)
|
|
60
|
-
|
|
61
|
-
x = F.relu(self.bn6(self.conv6(x)))
|
|
62
|
-
x = self.dropout6(x)
|
|
63
|
-
|
|
64
|
-
x = F.relu(self.bn7(self.conv7(x)))
|
|
65
|
-
x = self.dropout7(x)
|
|
66
|
-
|
|
67
|
-
x = F.relu(self.bn8(self.conv8(x)))
|
|
68
|
-
x = self.dropout8(x)
|
|
69
|
-
|
|
70
|
-
x = self.conv9(x)
|
|
71
|
-
|
|
72
|
-
return x
|
py3dcal-1.0.3.dist-info/RECORD
DELETED
|
@@ -1,44 +0,0 @@
|
|
|
1
|
-
py3DCal/__init__.py,sha256=NM8ftqmz739vw9q6l8Uo-teuwklAX3Smywf9NSwXSNA,755
|
|
2
|
-
py3DCal/data_collection/Calibrator.py,sha256=pxj6gqrQLHTxbMsdUfsUxUgOgY8pLqKarzkYxjpxW58,11496
|
|
3
|
-
py3DCal/data_collection/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
4
|
-
py3DCal/data_collection/Printers/Printer.py,sha256=ouqgWuJWk8PPjhTRFwolnupXbE0SzO819LIgw1ug-7s,1628
|
|
5
|
-
py3DCal/data_collection/Printers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
6
|
-
py3DCal/data_collection/Printers/Ender3/Ender3.py,sha256=pkE9Kt-mMH-fpZC5Gl6YyDPAxOdfZxp4Z7s68N7D_is,2239
|
|
7
|
-
py3DCal/data_collection/Printers/Ender3/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
8
|
-
py3DCal/data_collection/Sensors/Sensor.py,sha256=i-2BQENm3tezpE0ErpY057pQfeH5F8iufWMbfrRx8_s,801
|
|
9
|
-
py3DCal/data_collection/Sensors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
10
|
-
py3DCal/data_collection/Sensors/DIGIT/DIGIT.py,sha256=1Jc1qXwinsOFjJ6-QUoUcXOBFjG4M6ylnUjbWMe3NSc,1268
|
|
11
|
-
py3DCal/data_collection/Sensors/DIGIT/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
12
|
-
py3DCal/data_collection/Sensors/DIGIT/default.csv,sha256=OqjTE9b5uAEAxrHyjsxS7UyUWHgjO8yWPN0gVbBj7_Q,26728
|
|
13
|
-
py3DCal/data_collection/Sensors/GelsightMini/GelsightMini.py,sha256=1jr9nfpja_19gEflw_o0CGa9JC5Sz2PlhVZNzbBtMKM,1193
|
|
14
|
-
py3DCal/data_collection/Sensors/GelsightMini/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
15
|
-
py3DCal/data_collection/Sensors/GelsightMini/default.csv,sha256=lavPHcJ6o4VkvMvOk7lcdRCp9dOJxg_VrPNayf9zVvM,26449
|
|
16
|
-
py3DCal/model_training/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
17
|
-
py3DCal/model_training/datasets/DIGIT_dataset.py,sha256=IYRqWDawbTTe4IOVjZuKVr0yVNBe1XLGC6PoDxsTMfo,3017
|
|
18
|
-
py3DCal/model_training/datasets/GelSightMini_dataset.py,sha256=H8Fr_4f3HDHLLl6KshRfqt0FP8-3d4n9XRK0xfPcH0k,3070
|
|
19
|
-
py3DCal/model_training/datasets/__init__.py,sha256=vqrB177ZXrBmqDnL472EWleJS6Y-BxYEy2Ao9hWWDHc,137
|
|
20
|
-
py3DCal/model_training/datasets/split_dataset.py,sha256=AzNJlTgcXGa9AdHJnVJYNEyv__OuNHZAMB76Haqc-io,1351
|
|
21
|
-
py3DCal/model_training/datasets/tactile_sensor_dataset.py,sha256=O7jEtArQukV-jssXLHEueRiII5hE01kv2OBn0HS82Dc,3246
|
|
22
|
-
py3DCal/model_training/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
23
|
-
py3DCal/model_training/lib/add_coordinate_embeddings.py,sha256=8wit43RIx28IKEB82SnH_Of09FAmiWM3jgOeXpamM1I,1198
|
|
24
|
-
py3DCal/model_training/lib/depthmaps.py,sha256=v-UL7ui8ZMJLMtwVqfrYIecS2KTGjRwPXa7mhFQLztk,2684
|
|
25
|
-
py3DCal/model_training/lib/fast_poisson.py,sha256=wJ5MTkSCxkFU3wUx-zomvIYPcAyEpPZj-LX7JQOx8JE,2252
|
|
26
|
-
py3DCal/model_training/lib/get_gradient_map.py,sha256=IbCigrK_-6ZkeOSaHZAIhMu2pFmkSpWAaz1EjUtenCM,1438
|
|
27
|
-
py3DCal/model_training/lib/precompute_gradients.py,sha256=zc1uvishZP7PjBWYF2VSrIMCtEkLrTPtLktOTpCh9P8,1860
|
|
28
|
-
py3DCal/model_training/lib/train_model.py,sha256=fxFIfKWp3WA1Aa2IEczKBJCivVyVovj7IW2HqNw5IlE,4016
|
|
29
|
-
py3DCal/model_training/lib/validate_device.py,sha256=P_yFdAa0nj8camlEryol7AVtBos0gMhtalmAVLGwx08,778
|
|
30
|
-
py3DCal/model_training/lib/validate_parameters.py,sha256=lOOw5zg4feQngVElr54Q0SHsF7bhb7-3MFF_iceZNfE,1368
|
|
31
|
-
py3DCal/model_training/models/__init__.py,sha256=FTMTEyUglpRIFJ4hi3Z7y1KdScGDVVg_OxYXip4b8wg,42
|
|
32
|
-
py3DCal/model_training/models/touchnet.py,sha256=JIfhwXX0uIk6UcN2tmCLoiH9il9U2IByjbjYIjQEIqc,8079
|
|
33
|
-
py3DCal/model_training/touchnet/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
34
|
-
py3DCal/model_training/touchnet/dataset.py,sha256=L2n5mHa_90UeZSLluhDyS7yck5x2WfqaqRSouKFbBc4,2924
|
|
35
|
-
py3DCal/model_training/touchnet/touchnet.py,sha256=c7FyDviwclaAQtak-QY5710r53aiczhvvUAfoizYw24,31236
|
|
36
|
-
py3DCal/model_training/touchnet/touchnet_architecture.py,sha256=CGWodHCezCGVBPrjezj6574Sh0QQwSGWhRckUY8d7Hw,2137
|
|
37
|
-
py3DCal/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
38
|
-
py3DCal/utils/utils.py,sha256=hgTyWZuBXfo9lxLnOLd0445Aw2-uARtKGXuBhZmz-Z0,995
|
|
39
|
-
py3dcal-1.0.3.dist-info/LICENSE,sha256=D95ljbgz6PW9niwHP26EWFN77QBvepSCsMKGp0mRVFM,1066
|
|
40
|
-
py3dcal-1.0.3.dist-info/METADATA,sha256=vGiIuSBhT17_m0ciZhR066JNydbf0YDYZxVmgCbdA18,882
|
|
41
|
-
py3dcal-1.0.3.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
|
|
42
|
-
py3dcal-1.0.3.dist-info/entry_points.txt,sha256=_N1ruxvLEyZmSAaPsCx8kEzbYSJ5bHG5S8bvpua_X5E,59
|
|
43
|
-
py3dcal-1.0.3.dist-info/top_level.txt,sha256=NbatjyXjN_E6UMifZpkx-ohahGQH_ZFvqovwmvU7FMA,8
|
|
44
|
-
py3dcal-1.0.3.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|