我的哑代码:
ones = [1 for i in range(len(positivePictures))]
zeros = [0 for i in range(len(negativePictuers))]
y = zeros + ones
classes = np.array(y).reshape(1,len(y))
这有效,但花费的时间太长(大约 4K 图片(。我正在考虑使用
TF.ones和TF.Zeros相反。
但是我不知道如何将两个张量追加在一起。我知道我可以评估结果以返回一个 numpy 数组。但是如何附加它们呢?
UHHHHHG。 我讨厌回答我自己的问题。
def getOnesAndZeros(num_ones, num_zeros):
sess = tf.Session()
with sess.as_default():
ones = tf.ones(
(num_ones, 1),
dtype=tf.float32,
name=None
)
zeros = tf.zeros(
(num_zeros, 1),
dtype=tf.float32,
name=None
)
both = tf.concat([ones, zeros], 0)
npBoth = both.eval()
return npBoth.flatten()