Reshape errors when running the code

[oldraincoat]

您好，感谢您的分享。
我发现在运行代码时会出现一些错误。
在envs/carla/env.py 的35行，ENV_CONFIG["image_mode"] 被设置为"encode"。所以在346行的step(self, action)函数中：
obs = np.concatenate([self._image_rgb[-1], self.encode_measurement(info)], axis=2)
返回的obs的shape是[96, 96, 4]。
而在networks/conv_ha.py的166-167行中：
assert mean.shape[1:].as_list() == [96, 96, 3], mean.shape
mean = tf.reshape(mean, tools.shape(state)[:-1] + data_shape)
mean.shape[1:].as_list() == [96, 96, 3]，而data_shape == [96, 96, 4], 它们最后一维不相等，无法进行reshape操作。
期待您的回复，谢谢！