Create download_images.py

download_images.py

@@ -0,0 +1,34 @@
+import os
+import requests
+
+def download_images(url_list, output_folder):
+    # Kreiranje izlaznog foldera ako ne postoji
+    os.makedirs(output_folder, exist_ok=True)
+    
+    # Iteriranje kroz listu URL-ova
+    for i, url in enumerate(url_list):
+        try:
+            # Preuzimanje slike sa URL-a
+            response = requests.get(url)
+            if response.status_code == 200:
+                # Čuvanje slike
+                with open(os.path.join(output_folder, f"{i+1}.jpg"), "wb") as f:
+                    f.write(response.content)
+                print(f"Slika {i+1} uspešno preuzeta.")
+            else:
+                print(f"Greska prilikom preuzimanja slike {i+1}. Status kod: {response.status_code}")
+        except Exception as e:
+            print(f"Greska prilikom preuzimanja slike {i+1}: {str(e)}")
+
+# Lista URL-ova sa kojih ćemo preuzimati slike (primer)
+url_list = [
+    "https://example.com/image1.jpg",
+    "https://example.com/image2.jpg",
+    # Dodajte ostale URL-ove prema potrebi
+]
+
+# Folder u koji ćemo sačuvati preuzete slike
+output_folder = "dataset"
+
+# Pozivanje funkcije za preuzimanje slika
+download_images(url_list, output_folder)