Facial data input.py

import json
import requests
from PIL import Image, ImageDraw
import matplotlib.pyplot as plt
import os
from config import headers

# Run inference on an image
url = "https://api.ultralytics.com/v1/predict/R6nMlK6kQjSsQ76MPqQM"
data = {"size": 640, "confidence": 0.25, "iou": 0.45}
with open("data.jpg", "rb") as f:
    response = requests.post(url, headers=headers, data=data, files={"image": f})

# Check for successful response
response.raise_for_status()

data = response.json()

objects = data['data']

# 打开图片
image = Image.open('/test_data/ata.jpg')
image_backups = image
draw = ImageDraw.Draw(image)

# 在图片上标注结果并切割图片
for i, obj in enumerate(objects):
    if obj["name"] != "person":
        continue
    box = obj['box']
    cropped_image = image_backups.crop((box['x1'], box['y1'], box['x2'], box['y2']))
    plt.imshow(cropped_image)
    plt.show()
    name = input("请输入这个人的名字:")
    if name == "sb":
        continue
    if not os.path.exists(f'./face_data/{name}/'):
        os.mkdir(f'./face_data/{name}/')
        cropped_image.save(f'./face_data/{name}/1.jpg')


print("图片已经成功切割并保存到本地，同时在原图上标注了结果。")