l5kit.patch

--- l5kit/dataset/ego.py	2020-12-11 20:45:15.016839504 +0100
+++ ego.py	2020-12-11 20:47:41.250787310 +0100
@@ -97,7 +97,7 @@
             )
         data = self.sample_function(state_index, frames, self.dataset.agents, tl_faces, track_id)
         # 0,1,C -> C,0,1
-        image = data["image"].transpose(2, 0, 1)
+        image = data["image"] #.transpose(2, 0, 1)
 
         target_positions = np.array(data["target_positions"], dtype=np.float32)
         target_yaws = np.array(data["target_yaws"], dtype=np.float32)