def add_plot_2d(data, fig=None):
data = np.asarray(projectdims.project_cloud3dto2d(data))
ax4 = fig.gca()
# plot points in 3D
ax4.plot(data[:, 0], data[:, 1], '.')
plt.show()
return fig
def visualize_clouds_2d(filenum, new_path=None, point_left=None, point_right=None):
if new_path is not None:
path = new_path
fig=plt.figure()
ax = fig.gca()
for i in os.listdir(path):
fullpath = os.path.join(path, i)
if os.path.isfile(fullpath) and i.startswith(str(str(filenum) + '_')):
if 'skeleton' in fullpath:
drawskeleton_2d(fullpath, point_left, point_right, fig=fig, ax=ax)
else:
data = np.loadtxt(fullpath, delimiter=',')
if len(data) > 0:
data = np.asarray(projectdims.project_cloud3dto2d(data))
# plot points in 3D
ax.plot(data[:, 0], data[:, 1], '.')
plt.show()
