-
Notifications
You must be signed in to change notification settings - Fork 157
/
config.py
16 lines (14 loc) · 896 Bytes
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import os
TRAIN_DIR = 'flickr_logos_27_dataset'
IMAGES_DIR = os.path.join(TRAIN_DIR, 'flickr_logos_27_dataset_images')
CROPPED_IMAGES_DIR = os.path.join(TRAIN_DIR, 'flickr_logos_27_dataset_images_cropped')
ANNOT_FILE = os.path.join(TRAIN_DIR, 'flickr_logos_27_dataset_training_set_annotation.txt')
CROPPED_ANNOT_FILE = os.path.join(TRAIN_DIR, 'flickr_logos_27_dataset_training_set_annotation_cropped.txt')
CROPPED_ANNOT_FILE_TEST = os.path.join(TRAIN_DIR, 'flickr_logos_27_dataset_test_set_annotation_cropped.txt')
CLASS_NAMES = [
'Adidas', 'Apple', 'BMW', 'Citroen', 'Cocacola', 'DHL', 'Fedex', 'Ferrari',
'Ford', 'Google', 'HP', 'Heineken', 'Intel', 'McDonalds', 'Mini', 'Nbc',
'Nike', 'Pepsi', 'Porsche', 'Puma', 'RedBull', 'Sprite', 'Starbucks',
'Texaco', 'Unicef', 'Vodafone', 'Yahoo'
]
CLASS_NAMES_FILE = os.path.join(TRAIN_DIR, 'flickr_logos_27_dataset_class_names.txt')