arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.17845cs.CV

SPAC-Net:用于增强姿态估计的合成姿态感知动物ControlNet

SPAC-Net: Synthetic Pose-aware Animal ControlNet for Enhanced Pose Estimation

Northeastern University(东北大学)

Le Jiang, Sarah Ostadabbas

AI总结:

本文提出SPAC-Net,将ControlNet引入PASyn合成数据流程并设计Bi-ControlNet分别检测动物与背景边界,以生成更接近真实数据的带姿态标签合成图像,在AP10K上验证了无需真实数据即可训练高精度动物姿态估计网络的有效性。

摘要:

动物姿态估计已成为一个关键的研究领域,但标注数据的稀缺是开发精确模型的一个重大挑战。合成数据已成为一种有前景的替代方案,但它经常与真实数据存在领域差异。已有研究提出风格迁移算法来解决这一问题,但这些算法缺乏足够的空间对应性,导致标签信息丢失。在这项工作中,我们提出了一种名为合成姿态感知动物ControlNet(SPAC-Net)的新方法,该方法将ControlNet引入到先前提出的先验感知合成动物数据生成(PASyn)流程中。我们利用基于变分自编码器(VAE)的数据生成流程所生成的合理姿态数据,作为ControlNet整体嵌套边缘检测(HED)边界任务模型的输入,以生成姿态标签更接近真实数据的合成数据,从而无需真实数据即可训练高精度姿态估计网络。此外,我们提出了Bi-ControlNet结构,分别检测动物和背景的HED边界,提高了生成数据的精度和稳定性。使用SPAC-Net流程,我们生成了合成斑马和犀牛图像,并在AP10K真实数据集上进行了测试,结果表明其性能优于仅使用真实图像或其他方法生成的合成数据。我们的工作证明了合成数据在克服动物姿态估计中标注数据有限这一挑战方面的潜力。

英文摘要:

Animal pose estimation has become a crucial area of research, but the scarcity of annotated data is a significant challenge in developing accurate models. Synthetic data has emerged as a promising alternative, but it frequently exhibits domain discrepancies with real data. Style transfer algorithms have been proposed to address this issue, but they suffer from insufficient spatial correspondence, leading to the loss of label information. In this work, we present a new approach called Synthetic Pose-aware Animal ControlNet (SPAC-Net), which incorporates ControlNet into the previously proposed Prior-Aware Synthetic animal data generation (PASyn) pipeline. We leverage the plausible pose data generated by the Variational Auto-Encoder (VAE)-based data generation pipeline as input for the ControlNet Holistically-nested Edge Detection (HED) boundary task model to generate synthetic data with pose labels that are closer to real data, making it possible to train a high-precision pose estimation network without the need for real data. In addition, we propose the Bi-ControlNet structure to separately detect the HED boundary of animals and backgrounds, improving the precision and stability of the generated data. Using the SPAC-Net pipeline, we generate synthetic zebra and rhino images and test them on the AP10K real dataset, demonstrating superior performance compared to using only real images or synthetic data generated by other methods. Our work demonstrates the potential for synthetic data to overcome the challenge of limited annotated data in animal pose estimation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑