iDP3的Learning代码解析：逐步分解iDP3的数据集、模型、动作预测策略代码(包含2D和3D两个版本)
💡
原文中文，约19200字，阅读约需46分钟。
📝
内容提要
本文记录了作者与同事在2025年1月的出差经历，探讨了智能制造与大模型结合的趋势。详细介绍了diffusion_policy_3d的代码结构，包括数据集处理、模型构建和策略实现，强调了其在机器人和点云数据处理中的应用。
🎯
关键要点
- 作者与同事在2025年1月出差，探讨智能制造与大模型结合的趋势。
- 出差期间参观了多个城市的工厂，讨论智能升级和改造。
- 强调传统深度学习方法向大模型和模仿学习迁移的趋势。
- diffusion_policy_3d代码结构包括数据集处理、模型构建和策略实现。
- 数据集处理部分定义了低维、图像、点云和通用数据集的基类。
- GR1DexDataset3D类用于处理3D数据集，包含数据加载和验证集生成。
- GR1DexDatasetImage类用于处理图像和深度信息的数据集。
- 模型部分实现了交叉注意力机制、条件残差块和条件U-Net网络。
- EMAModel类实现了模型权重的指数移动平均，提高模型稳定性。
- SinusoidalPosEmb类用于为输入数据添加位置信息。
- MultiStagePointNetEncoder类用于对点云数据进行编码，提取全局特征。
- DiffusionImagePolicy类实现基于扩散模型的图像策略生成。
- DiffusionPointcloudPolicy类实现基于扩散模型的点云策略生成。
➡️
