论文: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‘
最终运行成功,结果如下,和论文中的结果一样