@@ -49,12 +49,12 @@ def run(self, ips, imgs, para = None):
4949 balls , ids , rs , graph = [], [], [], ips .data
5050 for idx in graph .nodes ():
5151 ids .append (idx )
52- balls .append (graph .node [idx ]['o' ])
52+ balls .append (graph .nodes [idx ]['o' ])
5353 xs , ys , zs = [], [], []
5454 lxs , lys , lzs = [], [], []
5555 for (s , e ) in graph .edges ():
5656 eds = graph [s ][e ]
57- st , ed = graph .node [s ]['o' ], graph .node [e ]['o' ]
57+ st , ed = graph .nodes [s ]['o' ], graph .nodes [e ]['o' ]
5858 lxs .append ([st [0 ],ed [0 ]])
5959 lys .append ([st [1 ],ed [1 ]])
6060 lzs .append ([st [2 ],ed [2 ]])
@@ -102,14 +102,14 @@ def run(self, ips, imgs, para = None):
102102 balls , ids , rs , graph = [], [], [], ips .data
103103 for idx in graph .nodes ():
104104 ids .append (idx )
105- balls .append (graph .node [idx ]['o' ])
105+ balls .append (graph .nodes [idx ]['o' ])
106106
107107
108108 xs , ys , zs = [], [], []
109109 v1s , v2s = [], []
110110 for (s , e ) in graph .edges ():
111111 eds = graph [s ][e ]
112- st , ed = graph .node [s ]['o' ], graph .node [e ]['o' ]
112+ st , ed = graph .nodes [s ]['o' ], graph .nodes [e ]['o' ]
113113 v1s .append (st )
114114 v2s .append (ed )
115115 for i in eds :
@@ -157,13 +157,13 @@ def run(self, ips, imgs, para = None):
157157 comid = 0
158158 for g in nx .connected_component_subgraphs (ips .data , False ):
159159 for idx in g .nodes ():
160- o = g .node [idx ]['o' ]
160+ o = g .nodes [idx ]['o' ]
161161 nodes .append ([comid , idx , g .degree (idx ), round (o [1 ]* k ,2 ), round (o [0 ]* k ,2 ), round (o [2 ])])
162162 for (s , e ) in g .edges ():
163163 eds = g [s ][e ]
164164 for i in eds :
165165 l = round (eds [i ]['weight' ]* k , 2 )
166- dis = round (np .linalg .norm (g .node [s ]['o' ]- g .node [e ]['o' ])* k , 2 )
166+ dis = round (np .linalg .norm (g .nodes [s ]['o' ]- g .nodes [e ]['o' ])* k , 2 )
167167 edges .append ([comid , s , e , l , dis ])
168168 comid += 1
169169
@@ -215,7 +215,7 @@ def run(self, ips, imgs, para = None):
215215 graph = ips .data
216216 datas = []
217217 for s in graph .nodes ():
218- o = graph .node [s ]['o' ]
218+ o = graph .nodes [s ]['o' ]
219219 x = graph [s ]
220220 if len (x )<= 1 : continue
221221 rst = []
@@ -302,8 +302,8 @@ def run(self, ips, imgs, para = None):
302302 if len (e1 )!= 1 or len (e2 )!= 1 : continue
303303 e1 , e2 = e1 [0 ], e2 [0 ]
304304 l1 , l2 = e1 ['pts' ], e2 ['pts' ]
305- d1 = norm (l1 [0 ]- g .node [n ]['o' ]) > norm (l1 [- 1 ]- g .node [n ]['o' ])
306- d2 = norm (l2 [0 ]- g .node [n ]['o' ]) < norm (l2 [- 1 ]- g .node [n ]['o' ])
305+ d1 = norm (l1 [0 ]- g .nodes [n ]['o' ]) > norm (l1 [- 1 ]- g .nodes [n ]['o' ])
306+ d2 = norm (l2 [0 ]- g .nodes [n ]['o' ]) < norm (l2 [- 1 ]- g .nodes [n ]['o' ])
307307 pts = np .vstack ((l1 [::[- 1 ,1 ][d1 ]], l2 [::[- 1 ,1 ][d2 ]]))
308308 l = np .linalg .norm (pts [1 :]- pts [:- 1 ], axis = 1 ).sum ()
309309 g .remove_node (n )
0 commit comments