File size: 1,066 Bytes
4fe19fe | 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 31 32 | import sys
import os
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
import argparse
import yaml
from data.dataset import CreatureDataset
def main():
parser = argparse.ArgumentParser(description="Pre-generate synthetic SMAL quadruped dataset.")
parser.add_argument("--config", type=str, default="configs/default_config.yaml")
args = parser.parse_args()
with open(args.config, "r", encoding="utf-8") as f:
cfg = yaml.safe_load(f)
cache_dir = cfg["data"]["cache_dir"]
samples_per_family = cfg["data"]["num_samples_per_family"]
num_points = cfg["data"]["num_points"]
print(f"Generating dataset: {samples_per_family} samples per family, {num_points} points...")
dataset = CreatureDataset(
num_samples_per_family=samples_per_family,
num_points=num_points,
augment=False,
cache_dir=cache_dir
)
print(f"Dataset successfully created and cached in {cache_dir}/ (Total: {len(dataset)} samples).")
if __name__ == "__main__":
main()
|