状态与动作的处理及设计
1. 状态处理
1.1 图像预处理代码
在处理图像数据时,通常需要进行一些预处理操作,以下是相关的代码示例:
import cv2
import numpy as np
def resize_image(im, w_h):
return cv2.resize(im, w_h, interpolation=cv2.INTER_AREA)
def normalize_image(im):
'''Normalizing image by dividing max value 255'''
return np.divide(im, 255.0)
def preprocess_image(im):
'''
Image preprocessing using OpenAI Baselines method: grayscale, resize
This resize uses stretching instead of cropping
'''
im = to_opencv_image(im)
im = grayscale_image(im)
im = resize_image(im, (84, 84))
im = np.expand_dims(im, 0)
return im
上述代码中, resize_image 函数用于调整图像的大小, normalize_image 函数用于将图像进行归一化处理,
超级会员免费看
订阅专栏 解锁全文

被折叠的 条评论
为什么被折叠?



