【人脸关键点检测】HIH-Towards More Accurate Face Alignment via Heatmap in Heatmap论文复现

论文:HIH-Towards More Accurate Face Alignment via Heatmap in Heatmap
代码:https://github.com/starhiking/HeatmapInHeatmap
环境:python3.8+代码中的requirements.txt里面要求的库
目前作者只发布了在WFLW数据集上面的评估代码和预训练模型,训练代码还没有发布,所以本篇文章仅针对评估代码运行的问题
在这里插入图片描述

运行中遇到的问题:
1、ModuleNotFoundError: No module named ‘torch.fx’

torch版本不匹配
解决办法:
#卸载pytorch
pip uninstall torch

再安装

python -m pip install torch -i https://mirrors.aliyun.com/pypi/simple/

2、ImportError: cannot import name ‘container_abcs’ from ‘torch._six’
解决办法:
ImportError: cannot import name ‘container_abcs‘ from ‘torch._six‘

最终运行成功,结果如下,和论文中的结果一样
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_43050258/article/details/127426569