最近在做人脸识别,老是对数据类型不清楚。数据本来在cuda上面,当本地要对数据进行操作时,需要将数据转到本地。一下是获取人的面部特征,并将其保存在字典中
model.encode(person[None, …]).cpu().numpy().reshape(-1).tolist()
data from cuda to cpu
最新推荐文章于 2023-04-07 23:24:31 发布
最近在做人脸识别,老是对数据类型不清楚。数据本来在cuda上面,当本地要对数据进行操作时,需要将数据转到本地。一下是获取人的面部特征,并将其保存在字典中
model.encode(person[None, …]).cpu().numpy().reshape(-1).tolist()