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

changed label mapping

Browse Source
This commit is contained in:
Markus Friedrich 2019-08-14 10:43:51 +02:00
parent 07043bd39b
commit 46334b8456
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pointcloud.py
View File

@ -205,6 +205,16 @@ def write_clusters(path, clusters, type_column=6):
types = np.unique(cluster[:, type_column], axis=0).astype(int) types = np.unique(cluster[:, type_column], axis=0).astype(int)
def type_mapping(t):
if t == 0:
return 2
elif t == 1:
return 1
elif t == 3:
return 4
return t
types = np.array([type_mapping(t) for t in types])
print("Types: ", types) print("Types: ", types)
np.savetxt(file, types.reshape(1, types.shape[0]),delimiter=';', header='', comments='', fmt='%i') np.savetxt(file, types.reshape(1, types.shape[0]),delimiter=';', header='', comments='', fmt='%i')