fixed bug with not setting training width and height and changed rotation range to 5 degrees x, y and 30 degrees z axis