未验证 提交 5b3bd165 编写于 作者: L lijianshe02 提交者: GitHub

fix fpgm prune criterion bug test=develop (#234)

上级 a13fe40b
...@@ -63,7 +63,7 @@ def geometry_median(group, graph): ...@@ -63,7 +63,7 @@ def geometry_median(group, graph):
x = w.repeat(channel_num, axis=0) x = w.repeat(channel_num, axis=0)
y = np.zeros_like(x) y = np.zeros_like(x)
for i in range(channel_num): for i in range(channel_num):
y[i * channel_num:(i + 1) * channel_num] = np.tile(channel_num, y[i * channel_num:(i + 1) * channel_num] = np.tile(w[i],
(channel_num, 1)) (channel_num, 1))
tmp = np.sqrt(np.sum((x - y)**2, -1)) tmp = np.sqrt(np.sum((x - y)**2, -1))
tmp = tmp.reshape((channel_num, channel_num)) tmp = tmp.reshape((channel_num, channel_num))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册