data from cuda to cpu

最近在做人脸识别,老是对数据类型不清楚。数据本来在cuda上面,当本地要对数据进行操作时,需要将数据转到本地。一下是获取人的面部特征,并将其保存在字典中
model.encode(person[None, …]).cpu().numpy().reshape(-1).tolist()


版权声明:本文为u014229742原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。