-
Notifications
You must be signed in to change notification settings - Fork 1
/
preprocess.py
42 lines (38 loc) · 1.02 KB
/
preprocess.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
import keras.preprocessing.image as kim
import cv2
def image_preprocess(image, brightness=1.0):
"""
Resizes one image, applies the needed brightness_adjust and converts to HSV colour space
Normalizes the data to [-1 to 1]
Parameters
----------
image : numpy array
The image
brightness: float
brightness adjusment factor
Returns
-------
image : numpy array
The processed image
"""
arr = image_brightness_adjust(image, brightness)
arr = cv2.resize(arr, (32, 16))
arr = arr / 127.5 - 1
return arr
def image_brightness_adjust(image, brightness):
"""
Convert an image to HSV colour space, apply brightness adjustment
Parameters
----------
image : numpy array
The image
brightness: float
brightness adjusment factor
Returns
-------
image : numpy array
The S channel of the image
"""
arr = cv2.cvtColor(image,cv2.COLOR_RGB2HSV)
arr[:, :, 2] = arr[:, :, 2] * brightness
return arr[:, :, 1]