File size: 687 Bytes
987ed1b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | if __name__ == "__main__":
import sys
import os
import pathlib
ROOT_DIR = str(pathlib.Path(__file__).parent.parent.parent)
sys.path.append(ROOT_DIR)
import click
import numpy as np
import json
from diffusion_policy.common.replay_buffer import ReplayBuffer
@click.command()
@click.option('--input', '-i', required=True)
@click.option('--dt', default=0.1, type=float)
def main(input, dt):
buffer = ReplayBuffer.create_from_path(input)
lengths = buffer.episode_lengths
durations = lengths * dt
result = {
'duration/mean': np.mean(durations)
}
text = json.dumps(result, indent=2)
print(text)
if __name__ == '__main__':
main()
|