| 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) |