在为 caffe 网络准备数据的脚本中,以下代码将图像(表示图像的 numpy 数组)转换为数据对象。
datum = caffe_pb2.Datum(
channels=3,
width=224,
height=224,
label=label,
data=np.rollaxis(img, 2).tostring())
如果网络是无监督的,是否只需以相同的方式创建对象但不填充标签参数,如下所示?
datum = caffe_pb2.Datum(
channels=3,
width=224,
height=224,
data=np.rollaxis(img, 2).tostring())
最佳答案
Datum
的label
为optional
:
optional int32 label = 5;
意思是oyu不必提供它。
旁注:
Datum
是一种数据结构,主要用于“Data”
输入层,严格来说它不是训练网络的一部分。
Caffe 使用 N 维张量 Blobs
存储网络的数据和参数。
关于machine-learning - caffe 中的 Datum 对象 - 无监督网络,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45937283/