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-07-21 10:35:41 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)