请问有机会发出训练代码吗？

gengyanlei / Pytorch_Retinaface

Retinaface pytorch face-pose-detect face-key-point-detect

MIT License

36 stars 7 forks source link

Closed wanyne-yyds closed 2 years ago

gengyanlei commented 2 years ago

@wanyne-yyds 没有机会了，代码有bug未修复。

wanyne-yyds commented 2 years ago

@wanyne-yyds 没有机会了，代码有bug未修复。

我在YOLOX训练出了能预测人脸与姿态角的，但会容易训崩。如果可以的话，可以看下你计算Loss那部分代码吗？

gengyanlei commented 2 years ago

我就是简单把HopeNet-pytorch的损失拿过来根据人脸的检测复用了一下。

wanyne-yyds commented 2 years ago

我就是简单把HopeNet-pytorch的损失拿过来根据人脸的检测复用了一下。

好的，谢谢哦