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()