1 回答

TA貢獻(xiàn)1816條經(jīng)驗(yàn) 獲得超4個(gè)贊
問題是labels具有形狀(),即零維(它是標(biāo)量),而您嘗試堆疊的所有稀疏張量都是一維的。您應(yīng)該制作一個(gè)label與框數(shù)據(jù)張量具有相同形狀的張量:
# Assuming all box data tensors have the same shape
box_data_shape = tf.shape(x['image/object/bbox/xmin'])
# Make label data
labels = tf.ones(box_data_shape, dtype=tf.float32)
除此之外,由于您正在解析單個(gè)示例,因此您的所有稀疏張量都應(yīng)該是一維且連續(xù)的,因此您可以將轉(zhuǎn)換保存為密集并只采用它們.values:
y_train = tf.stack([x['image/object/bbox/xmin'].values,
x['image/object/bbox/ymin'].values,
x['image/object/bbox/xmax'].values,
x['image/object/bbox/ymax'].values,
labels], axis=1)
添加回答
舉報(bào)