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

dataset fixing

Browse Source
This commit is contained in:
Si11ium
2020-06-19 15:37:44 +02:00
parent 49b373a8a1
commit b3c67bab40
3 changed files with 20 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
main_pipeline.py
View File

@ -42,9 +42,8 @@ def restore_logger_and_model(log_dir):
def predict_prim_type(input_pc, model):
input_data = dict(norm=torch.tensor(np.array([input_pc[:, 3:6]], np.float)),
pos=torch.tensor(input_pc[:, 0:3]),
y=np.zeros(input_pc.shape[0])
input_data = dict(norm=torch.tensor(np.array([input_pc[:, 3:6]], np.float)).unsqueeze(0),
pos=torch.tensor(input_pc[:, 0:3]).unsqueeze(0),
)
batch_to_data = BatchToData()
@ -71,7 +70,7 @@ if __name__ == '__main__':
input_pc = normalize_pointcloud(input_pc)
grid_clusters = cluster_cubes(input_pc, [1,1,1], 2048)
grid_clusters = cluster_cubes(input_pc, [1,1,1], 1024)
ps.init()