caspervk/nightr
1
0
Fork 0
Code Releases 1 Activity

implemented context-prediction interface in milostrats

Browse source
This commit is contained in:
Milo 2019-04-06 13:52:25 +02:00
parent 31012f4216
commit 33edbcc743
1 changed files with 15 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
server/nightr/strategies/miloStrats.py
View file

@ -1,24 +1,31 @@
import cv2 import cv2
from datetime import datetime, timedelta from datetime import datetime, timedelta
from pytz import timezone from pytz import timezone
from server.nightr.util import Context, Prediction
def camImgStrat(): def camImgStrat(context : Context) -> Prediction:
img = cv2.imread('night.jpg',0) img = cv2.imread('night.jpg',0)
average = img.mean(axis=0).mean(axis=0) average = img.mean(axis=0).mean(axis=0)
print(average) print(average)
p = Prediction()
if average < 100: if average < 100:
return 1.0 p.probability = 1.0
p.reasons.append('Image was dark')
else: else:
return 0.0 p.reasons.append('Image was light')
p.probability = 0.0
return p
def australiaStrat(context : Context) -> Prediction:
def australiaStrat():
australia = timezone('Australia/Melbourne') australia = timezone('Australia/Melbourne')
t = datetime.now().astimezone(australia) t = datetime.now().astimezone(australia)
hour = t.hour hour = t.hour
p = Prediction()
if hour > 22 or hour < 6: if hour > 22 or hour < 6:
return 1.0 p.probability = 1.0
p.reasons.append('It\'s day-time in Australia')
else: else:
return 0.0 p.probability = 0.0
p.reasons.append('It\'s night-time in Australia')
return p