Hoe to visulize curves as shown in Figure 7?

[cuge1995]

[tiangexiang]

Hi, visualization of curves is essentially the drawing of connections between the selected nodes in order. In this case, you can simply collect the flatten_cur parameters here at every step, and draw the connections following the collection order. flatten_cur carries indices that can be used to index the input xyz coordinates. Note that flatten_cur flattened the whole batch indices, and I suggest setting batch_size=1 before any visualizations for simplicity. We may add an easy-to-use visualization module for the curves in the future.