/DAViD-onnxrun

使用onnxruntime部署微软发布的DAViD:深度估计、表面法线估计和软前景分割,包含C++和Python两个版本的程序

Primary LanguagePython

DAViD是微软在7月发布的一个针对人体的深度估计、表面法线估计和软前景分割的多任务模型,文章中了iccv2025顶会,跟facebook的Sapiens的效果相当。 训练源码在 https://github.com/microsoft/DAViD

在做人体的深度估计或者表面法线估计时,先做前景分割,预测得到的mask,作为深度估计、表面法线估计的输入先验信息,效果会很好的。 模型是在人体数据集上训练的,因此在人像图片上的效果很好,测试时发现,如果输入一张不是人像的图片,深度估计预测结果很差。

onnx文件在百度云盘,链接: https://pan.baidu.com/s/1q6YVWmxmkGVJSggK8DWPgw 提取码: rmpa