#!/bin/bash

set -euxo pipefail

bazel run -c opt vision/data/tools/wheel_road_annotator:build_dataset_main -- \
   --answers_dir=s3cache://zoox-testdata/vision/wheel_road_annotator/answers4/ \
   --detection_lmdb_fn=s3cache://zoox-testdata/vision/wheel_road_annotator/wheel_road_annotator_detections_2_tr.lmdb \
   --output_fn /tmp/wheel_road_train_v4.lmdb \
   --html /tmp/wheel_road_train_v4.html \
   --num_samples 10 \
   --split_parts 10 \
   --split_start 0 \
   --split_end 10 \
   --sample_dx_size 0.2 \
   --sample_dy_size 0.2 \
   --sample_scale_min 0.95 \
   --sample_scale_max 1.25 \
   --alsologtostderr

bazel run -c opt vision/data/tools/wheel_road_annotator:build_dataset_main -- \
   --answers_dir=s3cache://zoox-testdata/vision/wheel_road_annotator/answers3/ \
   --detection_lmdb_fn=s3cache://zoox-testdata/vision/wheel_road_annotator/wheel_road_annotator_detections.lmdb \
   --output_fn /tmp/wheel_road_test_v2.lmdb \
   --html /tmp/wheel_road_test_v2.html \
   --split_parts 10 \
   --split_start 9 \
   --split_end 10 \
   --alsologtostderr
