download_images.py

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)