steffen/Labeling-Primitives-With-Point2Net
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Now with included sorting per cloud variation - Some korrektions

Browse Source
This commit is contained in:
Si11ium 2019-08-02 15:54:42 +02:00
parent 45d64e5b82
commit d291d49a2d
4 changed files with 1 additions and 2049 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dataset/shapenet.py
View File

@ -137,7 +137,7 @@ class CustomShapeNet(InMemoryDataset):
if self.collate_per_element: if self.collate_per_element:
datasets[data_folder].append(data) datasets[data_folder].append(data)
if self.has_variations: if self.has_variations:
cloud_variations[int(element.name.split('_')[0])].append(data) cloud_variations[int(os.path.split(element)[-1].split('_')[0])].append(data)
if not self.collate_per_element: if not self.collate_per_element:
if self.has_variations: if self.has_variations:
for variation in cloud_variations.keys(): for variation in cloud_variations.keys():

BIN
predict/checkpoint/seg_model_custom_249.pth
View File

Binary file not shown.

2048
predict/data/raw/predict/580_9/100_pc.xyz
View File

File diff suppressed because it is too large Load Diff

0
predict/data/raw/predict/no.headers
View File