视网膜面火炬版
由于机密性,它不是我模型的最佳版本
感谢Alvin Yang( )
这是68个地标检测的分支,预先训练的模型位于./out中
进行96个地标检测(请参阅其他分支)
该模型还预测了地标的被遮挡部分,如果不希望它们出现,可以将其隐藏。 基于RetinaFace
当前模型
mobileNet V1 + FPN +上下文模块+回归器1.6MB CPU〜10FPS GPU 50FPU
火车:(请参考dataloader.py更改文件位置)
python3 train.py -train该模型使用LS3D-W数据集,或将您的数据集更改为demo.pt/ demo.jpg(68 * 2张量)的格式
使用本地摄像头:
python3 video_detect.py(需要删除所有'cuda()',并在CPU本地运行)
评估模型:
python3 train.py -train错
2024-03-14 19:40:41
3.29MB
Python
1