此错误error: (-215) scn == 3 || scn == 4 in function cv::cvtColor
仅在我的脚本中,但是如果我在IPython shell中运行代码,一切都会很好。该代码用于减少使用openai-gym
运行的DQN(Deeq-Q网络)的尺寸,我不知道为什么这不在我的脚本中而是在我的shell中。
我添加了一条评论,指出错误所在。如果需要,可以添加完整的代码。
部分代码:
def preprocess(self,inp):
inp=cv2.resize(inp,(84, 110))
inp=cv2.cvtColor(inp, cv2.COLOR_BGR2GRAY)
inp=inp[26:110,:]
ret,obs=cv2.threshold(inp,1,255,cv2.THRESH_BINARY)
return np.reshape(obs,(84,84,1))
def Q_Learning(self,modell_path="Your path here"):
observation=self.env.reset()
for i in range(self.itertime):
observation=self.preprocess(observation)#!Error appers here!
if np.random.rand()<=self.random_move_prop:
action=np.random.randint(low=0,high=self.env.action_space.n)
else:
action=self.model.predict(observation)
action=np.argmax(action)
new_observation, reward, done, info=self.env.step(action)
new_observation=self.preprocess(new_observation)
self.storage.append([observation,action,reward,new_observation,done])
observation=new_observation
if done:
self.env.reset()