machine-learning - 有没有什么方法可以在 caffe 中添加一个新层来保持类型为 unsigned int 的权重？

Question

我想添加一个新的二进制卷积层来实现 xor-net使用二进制权重，问题是我不知道如何将权重保存为无符号整数类型。有没有办法改变caffe中图层的blobs_数据类型？

Answer 1

Bolb 数据类型在 caffe.proto 中定义。可以看到它有两个指针

repeated float data = 5 [packed = true];
repeated double double_data = 8 [packed = true];

这意味着 caffe 只能处理 float 或 double Blob 数据。如果你想要一个uint8 Blob，你必须修改caffe.proto并重新构建caffe。