From 5c88bf98d03d6a7c60df9d15592d6ebdd0752d55 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Wed, 19 Apr 2023 15:00:08 +0200
Subject: [PATCH] download images in PNG
---
worker_generic_training_dataset/utils.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/worker_generic_training_dataset/utils.py b/worker_generic_training_dataset/utils.py
index 245e587..47f2bfc 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
--
GitLab