├── KNN_CUDA-0.2-py3-none-any.whl ├── LICENSE ├── README.md ├── asserts ├── framework.png └── real.jpg ├── causal_conv1d-1.0.0+cu118torch1.13cxx11abiFALSE-cp39-cp39-linux_x86_64.whl ├── data_preprocessing ├── episodes.json ├── package_calvin.py ├── package_rlbench.py ├── preprocess_calvin_instructions.py ├── preprocess_rlbench_instructions.py ├── rearrange_rlbench_demos.py └── rerender_highres_rlbench.py ├── datasets ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── __init__.cpython-39.pyc │ ├── dataset_engine.cpython-38.pyc │ ├── dataset_engine.cpython-39.pyc │ ├── utils.cpython-38.pyc │ └── utils.cpython-39.pyc ├── dataset_calvin.py ├── dataset_engine.py └── utils.py ├── diffuser_actor ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ └── __init__.cpython-39.pyc ├── trajectory_optimization │ ├── DDPM.PY │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── __init__.cpython-39.pyc │ │ ├── diffuser_actor.cpython-38.pyc │ │ └── diffuser_actor.cpython-39.pyc │ ├── cfm.py │ └── diffuser_actor.py └── utils │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── __init__.cpython-39.pyc │ ├── clip.cpython-38.pyc │ ├── clip.cpython-39.pyc │ ├── encoder.cpython-38.pyc │ ├── encoder.cpython-39.pyc │ ├── layers.cpython-38.pyc │ ├── layers.cpython-39.pyc │ ├── multihead_custom_attention.cpython-38.pyc │ ├── multihead_flash_attention.cpython-38.pyc │ ├── position_encodings.cpython-38.pyc │ ├── position_encodings.cpython-39.pyc │ ├── resnet.cpython-38.pyc │ ├── resnet.cpython-39.pyc │ ├── utils.cpython-38.pyc │ └── utils.cpython-39.pyc │ ├── clip.py │ ├── converter.py │ ├── encoder.py │ ├── layers.py │ ├── multihead_custom_attention.py │ ├── multihead_flash_attention.py │ ├── position_encodings.py │ ├── resnet.py │ ├── tools │ ├── __pycache__ │ │ └── point_utils.cpython-38.pyc │ └── point_utils.py │ └── utils.py ├── engine.py ├── environment.yaml ├── evaluate_policy.py ├── insatll.md ├── main_trajectory.py ├── online_evaluation_rlbench └── eval_peract.sh ├── scripts ├── gnfactor_train.sh └── precise_train.sh ├── switch-cuda.sh ├── tasks └── 7_precise_tasks_location_bounds.json └── utils ├── __init__.py ├── __pycache__ ├── __init__.cpython-38.pyc ├── __init__.cpython-39.pyc ├── common_utils.cpython-38.pyc ├── common_utils.cpython-39.pyc └── utils_with_rlbench.cpython-38.pyc ├── common_utils.py ├── pytorch3d_transforms.py ├── utils_with_calvin.py └── utils_with_rlbench.py /KNN_CUDA-0.2-py3-none-any.whl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/KNN_CUDA-0.2-py3-none-any.whl -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/README.md -------------------------------------------------------------------------------- /asserts/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/asserts/framework.png -------------------------------------------------------------------------------- /asserts/real.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/asserts/real.jpg -------------------------------------------------------------------------------- /causal_conv1d-1.0.0+cu118torch1.13cxx11abiFALSE-cp39-cp39-linux_x86_64.whl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/causal_conv1d-1.0.0+cu118torch1.13cxx11abiFALSE-cp39-cp39-linux_x86_64.whl -------------------------------------------------------------------------------- /data_preprocessing/episodes.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/data_preprocessing/episodes.json -------------------------------------------------------------------------------- /data_preprocessing/package_calvin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/data_preprocessing/package_calvin.py -------------------------------------------------------------------------------- /data_preprocessing/package_rlbench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/data_preprocessing/package_rlbench.py -------------------------------------------------------------------------------- /data_preprocessing/preprocess_calvin_instructions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/data_preprocessing/preprocess_calvin_instructions.py -------------------------------------------------------------------------------- /data_preprocessing/preprocess_rlbench_instructions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/data_preprocessing/preprocess_rlbench_instructions.py -------------------------------------------------------------------------------- /data_preprocessing/rearrange_rlbench_demos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/data_preprocessing/rearrange_rlbench_demos.py -------------------------------------------------------------------------------- /data_preprocessing/rerender_highres_rlbench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/data_preprocessing/rerender_highres_rlbench.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/datasets/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/datasets/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/dataset_engine.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/datasets/__pycache__/dataset_engine.cpython-38.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/dataset_engine.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/datasets/__pycache__/dataset_engine.cpython-39.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/datasets/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/datasets/__pycache__/utils.cpython-39.pyc -------------------------------------------------------------------------------- /datasets/dataset_calvin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/datasets/dataset_calvin.py -------------------------------------------------------------------------------- /datasets/dataset_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/datasets/dataset_engine.py -------------------------------------------------------------------------------- /datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/datasets/utils.py -------------------------------------------------------------------------------- /diffuser_actor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/__init__.py -------------------------------------------------------------------------------- /diffuser_actor/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /diffuser_actor/trajectory_optimization/DDPM.PY: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/trajectory_optimization/DDPM.PY -------------------------------------------------------------------------------- /diffuser_actor/trajectory_optimization/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diffuser_actor/trajectory_optimization/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/trajectory_optimization/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/trajectory_optimization/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/trajectory_optimization/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /diffuser_actor/trajectory_optimization/__pycache__/diffuser_actor.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/trajectory_optimization/__pycache__/diffuser_actor.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/trajectory_optimization/__pycache__/diffuser_actor.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/trajectory_optimization/__pycache__/diffuser_actor.cpython-39.pyc -------------------------------------------------------------------------------- /diffuser_actor/trajectory_optimization/cfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/trajectory_optimization/cfm.py -------------------------------------------------------------------------------- /diffuser_actor/trajectory_optimization/diffuser_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/trajectory_optimization/diffuser_actor.py -------------------------------------------------------------------------------- /diffuser_actor/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/clip.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/clip.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/clip.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/clip.cpython-39.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/encoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/encoder.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/encoder.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/encoder.cpython-39.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/layers.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/layers.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/layers.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/layers.cpython-39.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/multihead_custom_attention.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/multihead_custom_attention.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/multihead_flash_attention.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/multihead_flash_attention.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/position_encodings.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/position_encodings.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/position_encodings.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/position_encodings.cpython-39.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/resnet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/resnet.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/resnet.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/resnet.cpython-39.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/__pycache__/utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/__pycache__/utils.cpython-39.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/clip.py -------------------------------------------------------------------------------- /diffuser_actor/utils/converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/converter.py -------------------------------------------------------------------------------- /diffuser_actor/utils/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/encoder.py -------------------------------------------------------------------------------- /diffuser_actor/utils/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/layers.py -------------------------------------------------------------------------------- /diffuser_actor/utils/multihead_custom_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/multihead_custom_attention.py -------------------------------------------------------------------------------- /diffuser_actor/utils/multihead_flash_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/multihead_flash_attention.py -------------------------------------------------------------------------------- /diffuser_actor/utils/position_encodings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/position_encodings.py -------------------------------------------------------------------------------- /diffuser_actor/utils/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/resnet.py -------------------------------------------------------------------------------- /diffuser_actor/utils/tools/__pycache__/point_utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/tools/__pycache__/point_utils.cpython-38.pyc -------------------------------------------------------------------------------- /diffuser_actor/utils/tools/point_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/tools/point_utils.py -------------------------------------------------------------------------------- /diffuser_actor/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/diffuser_actor/utils/utils.py -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/engine.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/environment.yaml -------------------------------------------------------------------------------- /evaluate_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/evaluate_policy.py -------------------------------------------------------------------------------- /insatll.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/insatll.md -------------------------------------------------------------------------------- /main_trajectory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/main_trajectory.py -------------------------------------------------------------------------------- /online_evaluation_rlbench/eval_peract.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/online_evaluation_rlbench/eval_peract.sh -------------------------------------------------------------------------------- /scripts/gnfactor_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/scripts/gnfactor_train.sh -------------------------------------------------------------------------------- /scripts/precise_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/scripts/precise_train.sh -------------------------------------------------------------------------------- /switch-cuda.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/switch-cuda.sh -------------------------------------------------------------------------------- /tasks/7_precise_tasks_location_bounds.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/tasks/7_precise_tasks_location_bounds.json -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/utils/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/utils/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/common_utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/utils/__pycache__/common_utils.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/common_utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/utils/__pycache__/common_utils.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/utils_with_rlbench.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/utils/__pycache__/utils_with_rlbench.cpython-38.pyc -------------------------------------------------------------------------------- /utils/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/utils/common_utils.py -------------------------------------------------------------------------------- /utils/pytorch3d_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/utils/pytorch3d_transforms.py -------------------------------------------------------------------------------- /utils/utils_with_calvin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/utils/utils_with_calvin.py -------------------------------------------------------------------------------- /utils/utils_with_rlbench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SanMumumu/FlowRAM/HEAD/utils/utils_with_rlbench.py --------------------------------------------------------------------------------