Update generate_hmb3.py

ARiSE-Lab · Jul 17, 2018 · 5d83a0e · 5d83a0e
1 parent 1a95f78
commit 5d83a0e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/generate_hmb3.py b/generate_hmb3.py
@@ -1,5 +1,7 @@
 '''
 Generate images and steering angles from hmb3.bag
+Modified from
+https://github.com/udacity/self-driving-car/blob/master/steering-models/evaluation/generator.py
 '''
 import argparse
 import rosbag
@@ -60,4 +62,4 @@ def gen(bag):
             csvcontent = []
             csvcontent.append(ts)
             csvcontent.append(steering)
-            writer.writerow(csvcontent)
+            writer.writerow(csvcontent)