def preprocess(img):
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
img = Image.fromarray(img)
img = Scale(120)(img)
img = ToTensor()(img)
img = img.view(1, 3, 120, 160)
img = img.cuda()
return img
def preprocess(state):
img = state.screen_buffer
depth = state.depth_buffer
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
img = Image.fromarray(img)
img = Scale(120)(img)
img = ToTensor()(img)
img = img.view(1, 3, 120, 160)
depth = Image.fromarray(depth)
depth = Scale(120)(depth)
depth = ToTensor()(depth)
depth = depth.view(1, 1, 120, 160)
img = torch.cat((img, depth), 1)
img = img.cuda()
return img
