RGB observation

[yunhaif]

Hi,

I find the code uses gray scale images as observations. Could you explain how you train with RGB images? What datasets did you use for generating the RGB visualization results in Figure 5 in the paper?

Thanks!