download images in PNG

worker_generic_training_dataset/utils.py

@@ -46,7 +46,7 @@ def download_image(element, folder: Path):
         try:
             image = iio.imread(build_image_url(element))
             cv2.imwrite(
-                str(folder / f"{element.image_id}.jpg"),
+                str(folder / f"{element.image_id}.png"),
                 cv2.cvtColor(image, cv2.COLOR_BGR2RGB),
             )
             break