#!/bin/bash

set -euxo pipefail

#DETECTION_TRAIN_LMDB_FN="s3cache://zoox-testdata/vision/datasets/zoox_detection_dataset_v10/train10.lmdb"
#DETECTION_TEST_LMDB_FN="s3cache://zoox-testdata/vision/datasets/zoox_detection_dataset_v10/test10.lmdb"
#DETECTION_TRAIN_LMDB_FN="/mnt/flashblade/multibox_data/zoox_detection_dataset_v11/train11.lmdb"
#DETECTION_TEST_LMDB_FN="/mnt/flashblade/multibox_data/zoox_detection_dataset_v11/test11.lmdb"

DETECTION_TRAIN_LMDB_FN="$HOME/data/train11_shuffled.lmdb"
DETECTION_TEST_LMDB_FN="$HOME/data/test11_shuffled.lmdb"

bazel run -c opt vision/data/tools/wheel_road_annotator:build_dataset_inpainting_main -- \
   --detection_lmdb_fn="$DETECTION_TRAIN_LMDB_FN" \
   --output_fn /tmp/wheel_road_train_inpainting_v1.lmdb \
   --html /tmp/wheel_road_train_inpainting_v1.html \
   --split_parts 10 \
   --split_start 0 \
   --split_end 10 \
   --size 500000 \
   --alsologtostderr

bazel run -c opt vision/data/tools/wheel_road_annotator:build_dataset_inpainting_main -- \
   --detection_lmdb_fn="$DETECTION_TEST_LMDB_FN" \
   --output_fn /tmp/wheel_road_test_inpainting_v1.lmdb \
   --html /tmp/wheel_road_test_inpainting_v1.html \
   --split_parts 10 \
   --split_start 0 \
   --split_end 10 \
   --size 10000 \
   --alsologtostderr
