Spaces:
Running
Running
File size: 875 Bytes
d015578 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
import argparse
import numpy as np
import matplotlib.pyplot as plt
def main():
# Input arguments control
pars = argparse.ArgumentParser(description='3D model visualization')
pars.add_argument('file', type=str, help='File txt path')
args = pars.parse_args()
visualize_3Dmodel(args.file)
def visualize_3Dmodel(input_file):
with open(input_file) as f:
lines = f.readlines()
model = []
for line in lines:
line = line[:-1] # Remove \n
line_split = line.split('|')
values = np.array(line_split, dtype=float)
model.append(values)
model = np.array(model)
model_xyz = model[:, 1:]
# Show model
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter(model_xyz[:, 0], model_xyz[:, 1], model_xyz[:, 2]+0.8)
plt.show()
if __name__ == '__main__':
main()
|