forked from tensorflow/models
-
Notifications
You must be signed in to change notification settings - Fork 19
/
number_plates.py
48 lines (43 loc) · 1.31 KB
/
number_plates.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
import datasets.fsns as fsns
DEFAULT_DATASET_DIR = '/home/anuj/tf-od-api/models/research/attention_ocr/python/datasets/data/number_plates'
DEFAULT_CONFIG = {
'name':
'number_plates', # you can change the name if you want.
'splits': {
'train': {
'size': 250, # change according to your own train-test split
'pattern': 'train.tfrecord'
},
'test': {
'size': 49, # change according to your own train-test split
'pattern': 'test.tfrecord'
}
},
'charset_filename':
'charset-labels.txt',
'image_shape': (200,200,3),#(max_width, max_height, 3),
'num_of_views':
1,
'max_sequence_length':
20, # TO BE CONFIGURED
'null_code':
43,
'items_to_descriptions': {
'image':
'A 3 channel color image.',
'label':
'Characters codes.',
'text':
'A unicode string.',
'length':
'A length of the encoded text.',
'num_of_views':
'A number of different views stored within the image.'
}
}
def get_split(split_name, dataset_dir=None, config=None):
if not dataset_dir:
dataset_dir = DEFAULT_DATASET_DIR
if not config:
config = DEFAULT_CONFIG
return fsns.get_split(split_name, dataset_dir, config)