├── .gitignore ├── README.md ├── adaflow ├── codecs │ └── imagecodecs_numcodecs.py ├── common │ ├── checkpoint_util.py │ ├── json_logger.py │ ├── normalize_util.py │ ├── pytorch_util.py │ ├── replay_buffer.py │ ├── sampler.py │ └── timestamp_accumulator.py ├── config │ ├── task │ │ └── libero_image_abs.yaml │ ├── train_adaflow_unet_image_workspace.yaml │ └── train_diffusion_unet_ddim_image_workspace.yaml ├── dataset │ ├── base_dataset.py │ └── robomimic_replay_image_dataset.py ├── env │ └── libero │ │ └── libero_image_wrapper.py ├── env_runner │ ├── base_image_runner.py │ └── libero_image_runner.py ├── gym_util │ ├── async_vector_env.py │ ├── multistep_wrapper.py │ ├── sync_vector_env.py │ ├── video_recording_wrapper.py │ └── video_wrapper.py ├── libero_extension │ └── envs │ │ ├── __init__.py │ │ └── env_wrapper.py ├── model │ ├── common │ │ ├── dict_of_tensor_mixin.py │ │ ├── lr_scheduler.py │ │ ├── module_attr_mixin.py │ │ ├── normalizer.py │ │ ├── rotation_transformer.py │ │ ├── shape_util.py │ │ └── tensor_util.py │ ├── diffusion │ │ ├── conditional_unet1d.py │ │ ├── conv1d_components.py │ │ ├── ema_model.py │ │ ├── mask_generator.py │ │ └── positional_embedding.py │ └── vision │ │ ├── crop_randomizer.py │ │ ├── model_getter.py │ │ └── multi_image_obs_encoder.py ├── policy │ ├── adaflow_unet_image_policy.py │ ├── base_image_policy.py │ └── diffusion_unet_image_policy.py ├── real_world │ ├── keystroke_counter.py │ ├── multi_camera_visualizer.py │ ├── multi_realsense.py │ ├── real_data_conversion.py │ ├── real_env.py │ ├── real_inference_util.py │ ├── realsense_config │ │ ├── 415_high_accuracy_mode.json │ │ └── 435_high_accuracy_mode.json │ ├── rtde_interpolation_controller.py │ ├── single_realsense.py │ ├── spacemouse.py │ ├── spacemouse_shared_memory.py │ └── video_recorder.py └── workspace │ ├── base_workspace.py │ ├── train_adaflow_unet_image_workspace.py │ └── train_diffusion_unet_image_workspace.py ├── eval.py ├── requirements.txt ├── scripts ├── concat_demos.py ├── convert_hdf5_to_jpg.py ├── convert_libero_data.sh ├── convert_libero_to_abs_action.py ├── convert_libero_to_abs_action_called.py ├── convert_libero_to_robomimic_format.py └── dataset_states_to_obs.py ├── toy_1d └── 1d.ipynb └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/README.md -------------------------------------------------------------------------------- /adaflow/codecs/imagecodecs_numcodecs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/codecs/imagecodecs_numcodecs.py -------------------------------------------------------------------------------- /adaflow/common/checkpoint_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/common/checkpoint_util.py -------------------------------------------------------------------------------- /adaflow/common/json_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/common/json_logger.py -------------------------------------------------------------------------------- /adaflow/common/normalize_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/common/normalize_util.py -------------------------------------------------------------------------------- /adaflow/common/pytorch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/common/pytorch_util.py -------------------------------------------------------------------------------- /adaflow/common/replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/common/replay_buffer.py -------------------------------------------------------------------------------- /adaflow/common/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/common/sampler.py -------------------------------------------------------------------------------- /adaflow/common/timestamp_accumulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/common/timestamp_accumulator.py -------------------------------------------------------------------------------- /adaflow/config/task/libero_image_abs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/config/task/libero_image_abs.yaml -------------------------------------------------------------------------------- /adaflow/config/train_adaflow_unet_image_workspace.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/config/train_adaflow_unet_image_workspace.yaml -------------------------------------------------------------------------------- /adaflow/config/train_diffusion_unet_ddim_image_workspace.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/config/train_diffusion_unet_ddim_image_workspace.yaml -------------------------------------------------------------------------------- /adaflow/dataset/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/dataset/base_dataset.py -------------------------------------------------------------------------------- /adaflow/dataset/robomimic_replay_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/dataset/robomimic_replay_image_dataset.py -------------------------------------------------------------------------------- /adaflow/env/libero/libero_image_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/env/libero/libero_image_wrapper.py -------------------------------------------------------------------------------- /adaflow/env_runner/base_image_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/env_runner/base_image_runner.py -------------------------------------------------------------------------------- /adaflow/env_runner/libero_image_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/env_runner/libero_image_runner.py -------------------------------------------------------------------------------- /adaflow/gym_util/async_vector_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/gym_util/async_vector_env.py -------------------------------------------------------------------------------- /adaflow/gym_util/multistep_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/gym_util/multistep_wrapper.py -------------------------------------------------------------------------------- /adaflow/gym_util/sync_vector_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/gym_util/sync_vector_env.py -------------------------------------------------------------------------------- /adaflow/gym_util/video_recording_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/gym_util/video_recording_wrapper.py -------------------------------------------------------------------------------- /adaflow/gym_util/video_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/gym_util/video_wrapper.py -------------------------------------------------------------------------------- /adaflow/libero_extension/envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/libero_extension/envs/__init__.py -------------------------------------------------------------------------------- /adaflow/libero_extension/envs/env_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/libero_extension/envs/env_wrapper.py -------------------------------------------------------------------------------- /adaflow/model/common/dict_of_tensor_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/common/dict_of_tensor_mixin.py -------------------------------------------------------------------------------- /adaflow/model/common/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/common/lr_scheduler.py -------------------------------------------------------------------------------- /adaflow/model/common/module_attr_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/common/module_attr_mixin.py -------------------------------------------------------------------------------- /adaflow/model/common/normalizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/common/normalizer.py -------------------------------------------------------------------------------- /adaflow/model/common/rotation_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/common/rotation_transformer.py -------------------------------------------------------------------------------- /adaflow/model/common/shape_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/common/shape_util.py -------------------------------------------------------------------------------- /adaflow/model/common/tensor_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/common/tensor_util.py -------------------------------------------------------------------------------- /adaflow/model/diffusion/conditional_unet1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/diffusion/conditional_unet1d.py -------------------------------------------------------------------------------- /adaflow/model/diffusion/conv1d_components.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/diffusion/conv1d_components.py -------------------------------------------------------------------------------- /adaflow/model/diffusion/ema_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/diffusion/ema_model.py -------------------------------------------------------------------------------- /adaflow/model/diffusion/mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/diffusion/mask_generator.py -------------------------------------------------------------------------------- /adaflow/model/diffusion/positional_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/diffusion/positional_embedding.py -------------------------------------------------------------------------------- /adaflow/model/vision/crop_randomizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/vision/crop_randomizer.py -------------------------------------------------------------------------------- /adaflow/model/vision/model_getter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/vision/model_getter.py -------------------------------------------------------------------------------- /adaflow/model/vision/multi_image_obs_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/model/vision/multi_image_obs_encoder.py -------------------------------------------------------------------------------- /adaflow/policy/adaflow_unet_image_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/policy/adaflow_unet_image_policy.py -------------------------------------------------------------------------------- /adaflow/policy/base_image_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/policy/base_image_policy.py -------------------------------------------------------------------------------- /adaflow/policy/diffusion_unet_image_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/policy/diffusion_unet_image_policy.py -------------------------------------------------------------------------------- /adaflow/real_world/keystroke_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/keystroke_counter.py -------------------------------------------------------------------------------- /adaflow/real_world/multi_camera_visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/multi_camera_visualizer.py -------------------------------------------------------------------------------- /adaflow/real_world/multi_realsense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/multi_realsense.py -------------------------------------------------------------------------------- /adaflow/real_world/real_data_conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/real_data_conversion.py -------------------------------------------------------------------------------- /adaflow/real_world/real_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/real_env.py -------------------------------------------------------------------------------- /adaflow/real_world/real_inference_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/real_inference_util.py -------------------------------------------------------------------------------- /adaflow/real_world/realsense_config/415_high_accuracy_mode.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/realsense_config/415_high_accuracy_mode.json -------------------------------------------------------------------------------- /adaflow/real_world/realsense_config/435_high_accuracy_mode.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/realsense_config/435_high_accuracy_mode.json -------------------------------------------------------------------------------- /adaflow/real_world/rtde_interpolation_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/rtde_interpolation_controller.py -------------------------------------------------------------------------------- /adaflow/real_world/single_realsense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/single_realsense.py -------------------------------------------------------------------------------- /adaflow/real_world/spacemouse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/spacemouse.py -------------------------------------------------------------------------------- /adaflow/real_world/spacemouse_shared_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/spacemouse_shared_memory.py -------------------------------------------------------------------------------- /adaflow/real_world/video_recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/real_world/video_recorder.py -------------------------------------------------------------------------------- /adaflow/workspace/base_workspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/workspace/base_workspace.py -------------------------------------------------------------------------------- /adaflow/workspace/train_adaflow_unet_image_workspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/workspace/train_adaflow_unet_image_workspace.py -------------------------------------------------------------------------------- /adaflow/workspace/train_diffusion_unet_image_workspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/adaflow/workspace/train_diffusion_unet_image_workspace.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/eval.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/concat_demos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/scripts/concat_demos.py -------------------------------------------------------------------------------- /scripts/convert_hdf5_to_jpg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/scripts/convert_hdf5_to_jpg.py -------------------------------------------------------------------------------- /scripts/convert_libero_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/scripts/convert_libero_data.sh -------------------------------------------------------------------------------- /scripts/convert_libero_to_abs_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/scripts/convert_libero_to_abs_action.py -------------------------------------------------------------------------------- /scripts/convert_libero_to_abs_action_called.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/scripts/convert_libero_to_abs_action_called.py -------------------------------------------------------------------------------- /scripts/convert_libero_to_robomimic_format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/scripts/convert_libero_to_robomimic_format.py -------------------------------------------------------------------------------- /scripts/dataset_states_to_obs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/scripts/dataset_states_to_obs.py -------------------------------------------------------------------------------- /toy_1d/1d.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/toy_1d/1d.ipynb -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hxixixh/AdaFlow/HEAD/train.py --------------------------------------------------------------------------------