├── .gitignore ├── README.md ├── modules ├── arx5_controller.py ├── arx5_env.py ├── arx5_zmq_client.py ├── pose_trajectory_interpolator.py ├── replay_buffer.py └── timestamp_accumulator.py ├── peripherals ├── keystroke_counter.py ├── multi_camera_visualizer.py ├── multi_uvc_camera.py ├── spacemouse_shared_memory.py ├── uvc_camera.py └── video_recorder.py ├── scripts └── eval_arx5.py ├── shared_memory ├── shared_memory_queue.py ├── shared_memory_ring_buffer.py ├── shared_memory_util.py └── shared_ndarray.py ├── umi_arx_environment.yaml └── utils ├── cv2_util.py ├── cv_util.py ├── interpolation_util.py ├── other_util.py ├── pose_repr_util.py ├── pose_util.py ├── real_inference_util.py └── usb_util.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | data -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/README.md -------------------------------------------------------------------------------- /modules/arx5_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/modules/arx5_controller.py -------------------------------------------------------------------------------- /modules/arx5_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/modules/arx5_env.py -------------------------------------------------------------------------------- /modules/arx5_zmq_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/modules/arx5_zmq_client.py -------------------------------------------------------------------------------- /modules/pose_trajectory_interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/modules/pose_trajectory_interpolator.py -------------------------------------------------------------------------------- /modules/replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/modules/replay_buffer.py -------------------------------------------------------------------------------- /modules/timestamp_accumulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/modules/timestamp_accumulator.py -------------------------------------------------------------------------------- /peripherals/keystroke_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/peripherals/keystroke_counter.py -------------------------------------------------------------------------------- /peripherals/multi_camera_visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/peripherals/multi_camera_visualizer.py -------------------------------------------------------------------------------- /peripherals/multi_uvc_camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/peripherals/multi_uvc_camera.py -------------------------------------------------------------------------------- /peripherals/spacemouse_shared_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/peripherals/spacemouse_shared_memory.py -------------------------------------------------------------------------------- /peripherals/uvc_camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/peripherals/uvc_camera.py -------------------------------------------------------------------------------- /peripherals/video_recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/peripherals/video_recorder.py -------------------------------------------------------------------------------- /scripts/eval_arx5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/scripts/eval_arx5.py -------------------------------------------------------------------------------- /shared_memory/shared_memory_queue.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/shared_memory/shared_memory_queue.py -------------------------------------------------------------------------------- /shared_memory/shared_memory_ring_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/shared_memory/shared_memory_ring_buffer.py -------------------------------------------------------------------------------- /shared_memory/shared_memory_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/shared_memory/shared_memory_util.py -------------------------------------------------------------------------------- /shared_memory/shared_ndarray.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/shared_memory/shared_ndarray.py -------------------------------------------------------------------------------- /umi_arx_environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/umi_arx_environment.yaml -------------------------------------------------------------------------------- /utils/cv2_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/utils/cv2_util.py -------------------------------------------------------------------------------- /utils/cv_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/utils/cv_util.py -------------------------------------------------------------------------------- /utils/interpolation_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/utils/interpolation_util.py -------------------------------------------------------------------------------- /utils/other_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/utils/other_util.py -------------------------------------------------------------------------------- /utils/pose_repr_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/utils/pose_repr_util.py -------------------------------------------------------------------------------- /utils/pose_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/utils/pose_util.py -------------------------------------------------------------------------------- /utils/real_inference_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/utils/real_inference_util.py -------------------------------------------------------------------------------- /utils/usb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/umi-arx/HEAD/utils/usb_util.py --------------------------------------------------------------------------------