Minor modification to img_augment.py: read shape
Minor modification to img_augment.py: read shape

LiuVII/Self-driving-RC-car,LiuVII/Self-driving-RC-car,LiuVII/Self-driving-RC-car,LiuVII/Self-driving-RC-car