machine-learning - 使用 brew 添加新的辅助方法会引发错误

Question

我尝试使用 brew 添加一个新的辅助函数，如下所示。

brew.Register(testOp)

当我尝试运行上述代码时，出现以下错误：

AttributeError: Helper testOp 已经存在。请更改您的助手名称。

请帮我解决这个错误。

注意： testOp在同一个文件中定义。

Answer 1

首先，我们必须定义函数（我们的示例函数只调用 Abs 运算符）：

def absolute_value(model, blob_in, blob_out, **kwargs):
    model.Abs(blob_in, blob_out)

然后，我们可以注册新的辅助函数：

brew.Register(absolute_value)

最后，我们将对其进行测试：

model = ModelHelper(name="test")

x = np.array([[-1, 2, 3], [1, -2, 3], [1, 2, -3]])
workspace.FeedBlob('x', x.astype(np.float32))

brew.absolute_value(model, 'x', 'out')

workspace.RunNetOnce(model.net)

blob = workspace.FetchBlob('out')
print(blob)

输出应该是：

[[1. 2. 3.]
 [1. 2. 3.]
 [1. 2. 3.]]