내일배움캠프 AI - TIL 29

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

 

 

 

내일배움캠프 AI트랙 29Day

 

 

오늘 배운 내용

머신러닝 학습 프로세스

1. 모델생성
2. 데이터 확보
3. 데이터 라벨링 : x좌표 y좌표를 그려줘야함.
4. 학습

 

 

# 가상환경 셋팅

python -m venv venv

source venv/bin/activate

 

 # install dependencies

pip install -qr https://raw.githubusercontent.com/ultralytics/yolov5/master/requirements.txt

 

 

# 서버에서 이미지 읽어서 하는 것

import torch

import cv2

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

imgs = ['https://ultralytics.com/images/zidane.jpg']  # batch of images

 

results = model(imgs)

 

print(results.xyxy[0], results.xyxy[0][0][0].item())  # img1 predictions (tensor)

print(results.pandas().xyxy[0])  # img1 predictions (pandas)

 

tmp_img = cv2.imread('zidane.jpg')

cv2.rectangle(tmp_img, (int(results.xyxy[0][0][0].item()), int(results.xyxy[0][0][1].item())), (int(results.xyxy[0][0][2].item()), int(results.xyxy[0][0][3].item())), (255,255,255))

cv2.imwrite('result.png', tmp_img)