-
Notifications
You must be signed in to change notification settings - Fork 0
/
readTrafficSigns2.py
75 lines (68 loc) · 2.91 KB
/
readTrafficSigns2.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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
# The German Traffic Sign Recognition Benchmark
#
# sample code for reading the traffic sign images and the
# corresponding labels
#
# example:
#
# trainImages, trainLabels = readTrafficSigns('GTSRB/Training')
# print len(trainLabels), len(trainImages)
# plt.imshow(trainImages[42])
# plt.show()
#
# have fun, Christian
import matplotlib.pyplot as plt
import csv
# function for reading the images
# arguments: path to the traffic sign data, for example './GTSRB/Training'
# returns: list of images, list of corresponding labels
def readTrafficSigns_train(rootpath):
'''Reads traffic sign data for German Traffic Sign Recognition Benchmark.
Arguments: path to the traffic sign data, for example './GTSRB/Training'
Returns: list of images, list of corresponding labels'''
images = [] # images
labels = [] # corresponding labels
# loop over all 42 classes
for c in range(33,36):
prefix = rootpath + '/' + format(c, '05d') + '/' # subdirectory for class
gtFile = open(prefix + 'GT-'+ format(c, '05d') + '.csv') # annotations file
gtReader = csv.reader(gtFile, delimiter=';') # csv parser for annotations file
# gtReader.next() # skip header
next(gtReader)
# loop over all images in current annotations file
for row in gtReader:
images.append(plt.imread(prefix + row[0])) # the 1th column is the filename
if row[7] == '33':
labels.append('0') # the 8th column is the label
elif row[7] == '34':
labels.append('1') # the 8th column is the label
elif row[7] == '35':
labels.append('2') # the 8th column is the label
gtFile.close()
return images, labels
# '/home/dataset/GTSRB/Testing/00000/GT-00000.csv'
def readTrafficSigns_test(rootpath):
images = [] # images
labels = [] # corresponding labels
gtFile = open(rootpath + '/' + 'GT-final_test.csv')
gtReader = csv.reader(gtFile, delimiter=';')
next(gtReader)
for row in gtReader:
if int(row[7]) == 33 or int(row[7]) == 34 or int(row[7]) == 35:
images.append(plt.imread(rootpath + '/' + row[0]))
# labels.append(str(int(row[7]-33)))
if row[7] == '33':
labels.append('0') # the 8th column is the label
elif row[7] == '34':
labels.append('1') # the 8th column is the label
elif row[7] == '35':
labels.append('2') # the 8th column is the label
return images, labels
if __name__ == "__main__":
train_images, train_labels = readTrafficSigns_train("/home/dataset/GTSRB/Training")
print(len(train_images))
print(len(train_labels))
test_images, test_labels = readTrafficSigns_test("/home/dataset/GTSRB/Testing")
print(len(test_images))
print(len(test_labels))
print(test_labels)