# Copyright (c) 2014-2016, NVIDIA CORPORATION.  All rights reserved.

# Datasets
TRAIN_FILE = 'train.txt'
TRAIN_DB = 'train_db'
VAL_FILE = 'val.txt'
VAL_DB = 'val_db'
TEST_FILE = 'test.txt'
TEST_DB = 'test_db'
MEAN_FILE_IMAGE = 'mean.jpg'
LABELS_COUNT_FILE = 'labels_count.txt'
LABELS_INDEX_FILE = 'labels_index.txt'

# Classification jobs
LABELS_FILE = 'labels.txt'
DEFAULT_BATCH_SIZE = 16

# Caffe Protocol Buffers
MEAN_FILE_CAFFE = 'mean.binaryproto'

# Color Palette attribute
COLOR_PALETTE_ATTRIBUTE = 'color_palette'

# Log file name
LOG_FILE_NAME = 'dataset.log'

SUPPORTED_EXTENSIONS = ('.png', '.jpg', '.jpeg', '.bmp', '.ppm', '.pgm')
