Looks like there are two places that are a bit slow
n=neighborhood(s, 1, mode='out')
and
so_segends=so_red_nodes[match(segends, svids)]
The latter (which takes 2-3s for a 60k node neuron like mvp2, is slow because it assumes svids is unsorted (it should actually be sorted). If we could do it in one pass it would take no time at all (<1ms).
Looks like there are two places that are a bit slow
and
The latter (which takes 2-3s for a 60k node neuron like mvp2, is slow because it assumes svids is unsorted (it should actually be sorted). If we could do it in one pass it would take no time at all (<1ms).