skimage画图并调整维度成功输入至卷积输入层中

背景：本人最近学习强化学习，想直接把工程任务上的状态情况使用skimage库制图再输入至卷积层中提取图像特征。
任务：1、完成skimage相关制图
2、把图像调整为卷积输入层合适的格式
#一、完成skimage相关制图

import matplotlib.pyplot as plt
from skimage import draw,data

self.img=np.zeros([self.width, self.width, 3], dtype=np.uint8)//制作底层图像

r1,c1=draw.ellipse(self.ex*self.width,self.ey*self.width,self.bad_rad*self.width,self.bad_rad*self.width)//画实心圆
draw.set_color(self.img,[r1,c1],[255,0,0])//给圆上色

rr, cc =draw.line(int(xnear[0]*self.width), int(xnear[1]*self.width), int(xnew[0]*self.width), int(xnew[1]*self.width))//画线
draw.set_color(self.img,[rr,cc],[255,165,0])

plt.imshow(self.img)
plt.show()//显示

以上就是使用skimage库的一些基本绘图操作

#二、把图像调整为卷积输入层合适的格式

state = torch.FloatTensor(state).permute(2,0,1).unsqueeze(0).to(device)//格式转换