machine-learning - Caffe Python Layer 用于输入，将数据类型设置为 uint8

Question

我正在尝试在咖啡厅中训练 CNN。我想要进行大量数据增强，因此我使用 "Python" 层作为输入，如建议的 here 所示。 .

但是，我从日志中看到 Caffe 对我的所有数据使用数据类型 float32。这确实很浪费，因为我只处理 8 位整数。有没有办法告诉caffe使用dtype='uint8'？

我尝试在设置顶部时对数据进行类型转换:

top[0].data[...] = someArray.astype(np.uint8, copy=False)

但这不起作用。

有什么建议吗？

Answer 1

据我所知，caffe 目前编译为仅支持 float32 或 float64。我想lmdb/leveldb数据可以存储在 uint8格式，但 caffe 在内部将其转换为 float32阅读后。
事实上，您的输入数据是 uint8并不意味着整个处理保持这种方式，在第一个卷积/内积层，数据与 float 相乘，并且不再保证保持 uint8 .
所以，我想你应该忍受输入层的一点空间浪费，并放弃转换为 uint8 .