Modified preprocessing
This commit is contained in:
parent
18ad080026
commit
59d53b5c8a
2 changed files with 3 additions and 4 deletions
|
@ -67,4 +67,4 @@ class Environment:
|
||||||
content = pickle.loads(r.content)
|
content = pickle.loads(r.content)
|
||||||
return content['state'], content['reward'], content['done'], content['info']
|
return content['state'], content['reward'], content['done'], content['info']
|
||||||
|
|
||||||
# env = Environment("127.0.0.1", 5000)
|
env = Environment("127.0.0.1", 5000)
|
||||||
|
|
|
@ -4,6 +4,7 @@ from flask import Flask
|
||||||
from flask import request
|
from flask import request
|
||||||
import pickle
|
import pickle
|
||||||
import json
|
import json
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
# Make it so that it doesn't log every HTTP request
|
# Make it so that it doesn't log every HTTP request
|
||||||
import logging
|
import logging
|
||||||
|
@ -53,9 +54,7 @@ class PongEnv(Environment):
|
||||||
frame = frame[34:194, 15:145] # Crops to shape (160, 130)
|
frame = frame[34:194, 15:145] # Crops to shape (160, 130)
|
||||||
# Downsample
|
# Downsample
|
||||||
frame = cv2.resize(frame, (80, 80), interpolation=cv2.INTER_AREA)
|
frame = cv2.resize(frame, (80, 80), interpolation=cv2.INTER_AREA)
|
||||||
# Normalize
|
return frame.astype(np.uint8)
|
||||||
frame = frame / 255
|
|
||||||
return frame
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue