#! /bin/python
import argparse

import bird_or_bicyle

if __name__ == '__main__':
  parser = argparse.ArgumentParser()
  parser.add_argument('--split', default="all")
  args = parser.parse_args()

  if args.split == 'all':
    for split in ['train', 'test', 'extras']:
      bird_or_bicyle.verify_dataset_integrity(split)
  else:
    bird_or_bicyle.verify_dataset_integrity(args.split)
