
TB_SUB_DIR = 'tb_logs'
RESULTS_FILE_NAME = 'results.pt'
model_args_file_name = 'args.pt'


# general target keys
TARGET_DICT_GLOBAL_COUNT = 'global_count_targets'
TARGET_DICT_TOTAL_NUM_OBJECTS = 'total_num_objects'
TARGET_DICT_BOXES = 'boxes'
TARGET_DICT_ALL_BOXES = 'all_boxes'
TARGET_DICT_SEGMENTATIONS = 'segmentations'
TARGET_DICT_DENSITY_MAPS = 'density_maps'
TARGET_DICT_ALL_DENSITY_MAPS = 'all_density_maps'
TARGET_DICT_CLASSES = 'classes'
TARGET_DICT_FEATURE_REPR = 'feature_repr'
TARGET_DICT_NUM_ELEMENTS = 'NUM_ELEMENTS'
TARGET_DICT_REF_IMGS = 'references'
TARGET_DICT_IS_ZERO_SHOT = 'is_zero_shot'
