|
import requests |
|
import os |
|
|
|
|
|
def download_images(urls, folder): |
|
os.makedirs(folder, exist_ok=True) |
|
for i, url in enumerate(urls): |
|
response = requests.get(url) |
|
with open(f"{folder}/image_{i}.jpg", "wb") as f: |
|
f.write(response.content) |
|
|
|
|
|
def identify_bears(images_folder): |
|
|
|
API_KEY = 'YOUR_API_KEY' |
|
url = "https://api.clarifai.com/v2/models/c0c0ac362b03416da06ab3fa36fb58e3/outputs" |
|
|
|
headers = { |
|
"Authorization": f"Key {API_KEY}", |
|
"Content-Type": "application/json", |
|
} |
|
|
|
image_files = os.listdir(images_folder) |
|
for image_file in image_files: |
|
with open(f"{images_folder}/{image_file}", "rb") as f: |
|
response = requests.post(url, headers=headers, json={"inputs": [{"data": {"image": {"base64": f.read().hex()}}}]}) |
|
|
|
data = response.json() |
|
concepts = data["outputs"][0]["data"]["concepts"] |
|
for concept in concepts: |
|
if concept["name"] in ["grizzly bear", "black bear"]: |
|
print(f"Image {image_file}: {concept['name']} - Probability: {concept['value']}") |
|
|
|
|
|
if __name__ == "__main__": |
|
urls = [ |
|
"URL_TO_BEAR_IMAGE_1", |
|
"URL_TO_BEAR_IMAGE_2", |
|
|
|
] |
|
download_images(urls, "images") |
|
identify_bears("images") |
|
|