diff --git a/worker_generic_training_dataset/utils.py b/worker_generic_training_dataset/utils.py index 245e5872444353d1c330c99b942adde99a8149ce..47f2bfc44638a46328c9f0551a6c034c69488fd9 100644 --- a/worker_generic_training_dataset/utils.py +++ b/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