有没有办法在caffe中添加一个新图层来保持权重哪个类型是unsigned int？

Question

我想添加一个新的二进制卷积层，用二进制权重实现xor-net，问题是我不知道如何将权重保存为unsigned int的类型。有没有办法在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。