Merge branch 'img' into 'main'

Image saver works

See merge request ii/collabschool22/genetic-bike-a!13

images/get_img.py

 import numpy as np
 import matplotlib.pyplot as plt
-
+import json
+import os
 def save_bike_image(points_position, counter):
     """This function save an image of the bike.
 
@@ -11,8 +12,8 @@ def save_bike_image(points_position, counter):
     fig, ax = plt.subplots()
     ax.set_title(f"Bike {counter}")
     ax.axis('equal')
-    ax.set_xlim(0, 35)
-    ax.set_ylim(-5, 20)
+    ax.set_xlim(0, 15)
+    ax.set_ylim(-5, 10)
 
     num_point=4
     wheel_radius = 1.2
@@ -41,4 +42,18 @@ def save_bike_image(points_position, counter):
 
 #num_point=4
 #points_position = np.random.rand(num_point, 2)*8+3
-#save_bike_image(points_position, 2)
\ No newline at end of file
+#save_bike_image(points_position, 2)
+
+list_files = os.listdir('../data')
+for index,item in enumerate(list_files):
+    os.makedirs(f'population_{index}', exist_ok=True)
+    os.chdir(f'population_{index}')
+    with open(f'../../data/{item}', 'r') as f:
+        bikes = json.load(f)
+        print(len(bikes))
+        f.close()
+        for bike in range(len(bikes)):
+            print(bike)
+            save_bike_image(np.array(bikes[bike]),bike)
+    os.chdir('../')
+#print(type(bikes[0][0]))
\ No newline at end of file