221013 TIL

👩🏻‍💻 아침에 머신러닝 쪽지시험을 봤다. 어제 배운 특강에서 조금만 활용하면 되는 문제였다. 사실 왜 되는건지 어떻게 되는건지는 잘 모르겠지만.....뭔가 신기하긴 했다. ㅎㅎㅎㅎ이거 머신러닝 괜찮은건가.....

 

 

✅ 쪽지시험 ✅

import ssl
ssl._create_default_https_context = ssl._create_unverified_context
import torch
import cv2
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)

img = cv2.imread('picture.jpg')
results = model(img)
results.save()

result = results.pandas().xyxy[0].to_numpy()
result = [item for item in result if item[6]=='person']

tmp_img = cv2.imread('picture.jpg')
print(tmp_img.shape)

for i in range(len(result)):
    cropped = tmp_img[int(result[i][1]):int(result[i][3]), int(result[i][0]):int(result[i][2])]
    print(cropped.shape)
    cv2.imwrite(f'people{i}.png', cropped)