내일배움캠프 AI - TIL 30

❤️‍🔥TIL : Today I Learned❤️‍🔥
그날그날 내가 공부한 것을 정리하는 것

 

 

 

내일배움캠프 AI트랙 30Day

 

 

오늘 배운 내용

아침쪽지시험  4차 - 사물인식

1. 위 이미지를 다운받아 저장하세요
2. opencv 로 이미지를 읽고 이미지의 가로, 세로가 각 몇 pixel 인지 구하세요
   1. (세로, 가로)
3. 이미지에서 사람을 찾아 하얀색으로 네모를 그려서 result1.png 로 저장하세요
4. 이미지에서 사람들을 잘라 people1.png, people2.png… 로 저장하세요
5. 코드와 이미지를 git에 업로드하고 해당 repository를 공유해주세요

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

img = cv2.imread('sample.jpeg')
results = model(img)
h, w, c = img.shape
print(f'width:  {w}px, height: {h}px')

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

tmp_img = cv2.imread('sample.jpeg')
for i, item in enumerate(result):
    cropped = img[int(item[1]):int(item[3]), int(item[0]):int(item[2])]
    cv2.imwrite(f'person{i+1}.png', cropped)
    cv2.rectangle(tmp_img, (int(item[0]), int(item[1])), (int(item[2]), int(item[3])), (255,255,255))
cv2.imwrite('result1.png', tmp_img)