Moblie-Aloha家务机器人-2.项目的目录及内容

1、asset                  机械臂模型的三维文件及仿真环境的建模文件

2、byol_pytorch           与vinn相关的byol包

3、detr                   动作生成的主要模型文件

4、aligh.py               对齐,将两个机械臂的姿态对齐(对称)

5、command.txt            提供的可执行脚本

6、compress_data.py       对图像进行压缩,主要针对图像数据(JPEG)

7、constant.py            包含了任务配置与机械臂的姿态常数。

8、dxl_test.py            对移动底盘的测试文件

9、dynamixel_client.py    移动底盘的操作接口

10、ee_sim_env.py         两个任务的仿真环境的配置

11、imitate_episodes.py   主要的运行文件

12、policy.py             策略文件,包括了ACT,Diffusion,CNNMLP

13、postprocess_episodes.py      数据的后处理

14、record_sim_episodes.py       采集仿真环境下的训练数据

15、replay_episodes.py           重播采集的数据并存储视频

16、scripted_policy.py           为record_sim_episodes.py提供动作策略

17、sim_env.py                   仿真环境的创建

18、train_actuator_network.py    执行器网络,指的是移动底盘的控制网络

19、train_latent_model.py         与vq参数相关(vector-quantisation)

20、truncate.py                   对数据进行截断

21、vinn_cache_feature.py         vinn中的特征库建立

22、vinn_select.py                vinn算法的训练选择k个临近特征

23、vinn_eval.py                  vinn算法的验证

24、visualize_episodes.py         数据集的可视化

硬件代码结构2.png