boxes = boxes.cpu() scores = conf.squeeze(0).data.cpu()[:, 1] landms = decode_landm(landms.data.squeeze(0), prior_data, cfg[variance]) landms = landms * 640 landms = landms.cpu()
将下面代码中的numpy操作转换为纯pytorch函数:boxes = boxes.cpu().numpy() scores = conf.squeeze(0).data.cpu().numpy()[...
最新推荐文章于 2023-03-13 10:39:18 发布