diff options
Diffstat (limited to 'nicla/traffic_light.py')
-rw-r--r-- | nicla/traffic_light.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/nicla/traffic_light.py b/nicla/traffic_light.py index 9499aea..f445690 100644 --- a/nicla/traffic_light.py +++ b/nicla/traffic_light.py @@ -32,6 +32,7 @@ def rgb2hsv(rgb): def traf_lights(imgTraffic): + original = imgTraffic.copy(copy_to_fb=True) img = imgTraffic.to_grayscale() for blob in img.find_blobs([(0, 60)], pixels_threshold=100): aspect = blob.h() / blob.w() @@ -62,10 +63,10 @@ def traf_lights(imgTraffic): #print(("", "rood", "geel", "groen")[light_status]) if light_status == 1: - return 0x06 - elif light_status == 2: return 0x07 - elif light_status == 3: + elif light_status == 2: return 0x08 + elif light_status == 3: + return 0x09 else: return 0x01 |