np.concatenate()
np.concatenate()的用法和torch.cat()的用法相同
qf.append(features)
pids = pids.numpy().astype(np.int16)
camids = camids.numpy().astype(np.int16)
q_pids.append(pids)
q_camids.append(camids)
qf = torch.cat(qf)
q_pids = np.asarray(np.concatenate(q_pids))
q_camids = np.asarray(np.concatenate(q_camids))
其中q_pids包含的是numpy数组的列表,通过np.concatenate()可以直接转化成为列数组。如果先前不用pids = pids.numpy().astype(np.int16)来将pids转化为numpy数组,用torch.cat()也可以达到同样的效果。
评论
发表评论