Files
masks_augments_compare-21/variables.py
2020-12-17 08:02:29 +01:00

26 lines
688 B
Python

# Labels
from argparse import Namespace
CLEAR = 0
MASK = 1
# Task Options
TASK_OPTION_multiclass = 'multiclass'
N_CLASS_multi = 10
multi_classes_names = ['air_conditioner', 'car_horn', 'children_playing',
'dog_bar', 'drilling', 'engine_idling',
'gun_shot', 'jackhammer', 'siren', 'street_music']
multi_classes = {key: val for val, key in enumerate(multi_classes_names)}
TASK_OPTION_binary = 'binary'
N_CLASS_binary = 2
binary_CLASS_clear = 0
binary_CLASS_maske = 1
# Dataset Options
DATA_OPTION_test = 'test'
DATA_OPTION_devel = 'devel'
DATA_OPTION_train = 'train'
DATA_OPTIONS = [DATA_OPTION_train, DATA_OPTION_devel, DATA_OPTION_test]