使用test.py后，face alignment提示报错

minivision-ai / photo2cartoon

人像卡通化探索项目 (photo-to-cartoon translation project)

MIT License

3.95k stars 764 forks source link

使用test.py后，face alignment提示报错 #41

Closed shenghanqin closed 4 years ago

shenghanqin commented 4 years ago

我是Python新手，我在使用 test.py 时碰到了问题。一直卡在 self.fa = face_alignment.FaceAlignment(face_alignment.LandmarksType._2D, flip_input=False)这一句上，我猜测的原因为pytorch的模型没加载好。

shenghanqin commented 4 years ago

@hao-qiang 已确认安装了对应的版本

pip install torch==1.4.0 torchvision==0.5.0

hao-qiang commented 4 years ago

这个是face-alignment库下载模型不成功导致的，关于关键点检测模型下载问题可以参考 #3