CarDetection-Keras/create_training_data.py at master · ashislaha/CarDetection-Keras · GitHub

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
76
77

# import classes and functions

from PIL import Image
import matplotlib.pyplot as plot
import os,sys
from keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img


# Training Data
pathTrainDataCar = 'TrainImages/car/'
pathTrainDataBuilding = 'TrainImages/building/'
pathTrainDataRoad = 'TrainImages/road/'
pathTrainDataRandom = 'TrainImages/random/'
pathTestData = 'TrainImages/validation/'

pathResizedTrainDataCar = 'ResizeTrainImages/car/'
pathResizedTrainDataBuilding = 'ResizeTrainImages/building/'
pathResizedTrainDataRoad = 'ResizeTrainImages/road/'
pathResizedTrainDataRandom = 'ResizeTrainImages/random/'
pathResizedTestData = 'ResizeTrainImages/validation/'

# Resize the images
row,column = 100,100

listingCar = os.listdir(pathTrainDataCar)
print(listingCar)
for file in listingCar:
	if file != '.DS_Store' :
		img = Image.open(pathTrainDataCar + file)
		resizeImg = img.resize((row,column))
		gray = resizeImg.convert('L')
		gray.save(pathResizedTrainDataCar + file)


listingBuilding = os.listdir(pathTrainDataBuilding)
print(listingBuilding)
for file in listingBuilding:
	if file != '.DS_Store':
		img = Image.open(pathTrainDataBuilding + file)
		resizeImg = img.resize((row,column))
		gray = resizeImg.convert('L')
		gray.save(pathResizedTrainDataBuilding + file)


listingRoad = os.listdir(pathTrainDataRoad)
print(listingRoad)
for file in listingRoad:
	if file != '.DS_Store':
		img = Image.open(pathTrainDataRoad + file)
		resizeImg = img.resize((row,column))
		gray = resizeImg.convert('L')
		gray.save(pathResizedTrainDataRoad + file)


listingRandom = os.listdir(pathTrainDataRandom)
print(listingRandom)
for file in listingRandom:
	if file != '.DS_Store':
		img = Image.open(pathTrainDataRandom + file)
		resizeImg = img.resize((row,column))
		gray = resizeImg.convert('L')
		gray.save(pathResizedTrainDataRandom + file)


listingTestData = os.listdir(pathTestData)
print(listingTestData)
for file in listingTestData:
	if file != '.DS_Store':
		img = Image.open(pathTestData+file)
		resizeImage = img.resize((row,column))
		gray = resizeImage.convert('L')
		gray.save(pathResizedTestData+file)