Spaces:
Runtime error
Runtime error
| 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: | |
| # Extrakcija naziva datoteke iz URL-a | |
| filename = url.split("/")[-1] | |
| # Razdvajanje naziva datoteke na osnovu nota | |
| note = filename.split("_")[0] | |
| # Kreiranje podfoldera za notu ako ne postoji | |
| note_folder = os.path.join(output_folder, note) | |
| os.makedirs(note_folder, exist_ok=True) | |
| # Čuvanje slike u odgovarajućem podfolderu | |
| with open(os.path.join(note_folder, filename), "wb") as f: | |
| f.write(response.content) | |
| print(f"Slika {i+1} uspešno preuzeta i sačuvana u {note_folder}.") | |
| 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/do_image1.jpg", | |
| "https://example.com/re_image1.jpg", | |
| "https://example.com/mi_image1.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) |