steffen/point_to_primitive
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

DataSet Modifications & Checks

Browse Source
This commit is contained in:
Si11ium
2020-07-02 08:58:02 +02:00
parent 3c1202d5b6
commit e9d0591b11
3 changed files with 28 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
models/point_net_2.py
View File

@@ -25,7 +25,7 @@ class PointNet2(BaseValMixin,
# Dataset
# =============================================================================
# rot_max_angle = 15
trans_max_distance = 0.02
trans_max_distance = 0.01
transforms = Compose(
[
RandomFlip(0, p=0.8),
@@ -54,7 +54,6 @@ class PointNet2(BaseValMixin,
self.n_classes = len(GlobalVar.classes) if not self.params.poly_as_plane else (len(GlobalVar.classes) - 2)
# Modules
self.point_net_core = ()
self.lin3 = torch.nn.Linear(128, self.n_classes)
# Utility