File size: 1,126 Bytes
60a0db4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import pytz
import datetime
import os
import requests
from urllib.parse import urlparse
import time
from ultralytics import YOLO
def process_images(image_urls):
hong_kong_timezone = pytz.timezone('Asia/Hong_Kong')
while True:
current_time = datetime.datetime.now(tz=hong_kong_timezone).strftime("%Y%m%d%H%M%S")
folder_name = f"/content/{current_time}"
print(folder_name)
os.makedirs(folder_name, exist_ok=True)
for image_url in image_urls:
response = requests.get(image_url)
image_data = response.content
parsed_url = urlparse(image_url)
image_name = os.path.basename(parsed_url.path)
file_name = os.path.join(folder_name, image_name)
with open(file_name, "wb") as file:
file.write(image_data)
print(file_name)
folder_name_formatted = f"'{folder_name}'"
yolo = YOLO('/content/Smart-Traffic/best.pt')
yolo.set_conf(0.45)
yolo.predict(source=folder_name_formatted, save=True, save_txt=True)
time.sleep(120) |