kataniccc commited on
Commit
f147395
1 Parent(s): 613bcda

Create download_images.py

Browse files
Files changed (1) hide show
  1. download_images.py +34 -0
download_images.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import requests
3
+
4
+ def download_images(url_list, output_folder):
5
+ # Kreiranje izlaznog foldera ako ne postoji
6
+ os.makedirs(output_folder, exist_ok=True)
7
+
8
+ # Iteriranje kroz listu URL-ova
9
+ for i, url in enumerate(url_list):
10
+ try:
11
+ # Preuzimanje slike sa URL-a
12
+ response = requests.get(url)
13
+ if response.status_code == 200:
14
+ # Čuvanje slike
15
+ with open(os.path.join(output_folder, f"{i+1}.jpg"), "wb") as f:
16
+ f.write(response.content)
17
+ print(f"Slika {i+1} uspešno preuzeta.")
18
+ else:
19
+ print(f"Greska prilikom preuzimanja slike {i+1}. Status kod: {response.status_code}")
20
+ except Exception as e:
21
+ print(f"Greska prilikom preuzimanja slike {i+1}: {str(e)}")
22
+
23
+ # Lista URL-ova sa kojih ćemo preuzimati slike (primer)
24
+ url_list = [
25
+ "https://example.com/image1.jpg",
26
+ "https://example.com/image2.jpg",
27
+ # Dodajte ostale URL-ove prema potrebi
28
+ ]
29
+
30
+ # Folder u koji ćemo sačuvati preuzete slike
31
+ output_folder = "dataset"
32
+
33
+ # Pozivanje funkcije za preuzimanje slika
34
+ download_images(url_list, output_folder)