Spaces:
Runtime error
Runtime error
File size: 1,139 Bytes
9be6f3b |
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 39 40 |
from stl import mesh
from mpl_toolkits import mplot3d
from matplotlib import pyplot as plt
# Load the STL file
your_mesh = mesh.Mesh.from_file('sample_data.stl')
# Define three different viewing angles
viewing_angles = [(30, 45), (60, 90), (45, 135)]
# Iterate over each viewing angle and generate an image
for i, (elev, azim) in enumerate(viewing_angles, start=1):
# Create a new plot with a larger figure size
fig = plt.figure(figsize=(10, 10))
ax = fig.add_subplot(111, projection='3d')
# Add the STL file to the plot
ax.add_collection3d(mplot3d.art3d.Poly3DCollection(your_mesh.vectors))
# Calculate the limits of the mesh
max_dim = max(your_mesh.points.flatten())
min_dim = min(your_mesh.points.flatten())
# Set the limits of the plot
ax.set_xlim([min_dim, max_dim])
ax.set_ylim([min_dim, max_dim])
ax.set_zlim([min_dim, max_dim])
# Set the viewing angle
ax.view_init(elev=elev, azim=azim)
# Save the plot as an image
plt.savefig(f'mesh_{i}.png')
# Close the plot to avoid memory leaks
plt.close()
# Optional: Show the last generated plot
# plt.show()
|