딥러닝 공부 2024.05.20

딥러닝 개념

수학적 공식 발표 증빙하는 알고리즘 창조하는 영역이 하나가 있고

기존의 알고리즘을 가지고 사회에 접목(산업현장에 응용..)시키는 영역이 하나가 있음

cnn 이미지 rnn 동영상

yolo 개념

간단한 코드 구현

from ultralytics import YOLO
import cv2
from matplotlib import pyplot as plt

# YOLO 모델 로드
model = YOLO('yolov8n.pt')

# COCO 128 데이터셋으로 모델 학습
model.train(data='coco128.yaml', epochs=5)

# 학습된 모델로 예측
results = model.predict('person.jpg')

for i, result in enumerate(results):
    # 바운딩 박스 그리기
    img = result.plot()

    # 파일 저장 경로
    save_path = f'C:/Users/user123/Desktop/learn-py/yolo/05-22/prediction_{i}.jpg'
    cv2.imwrite(save_path, img)

    # 이미지를 BGR에서 RGB로 변환
    img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
    
    # 이미지 표시
    plt.imshow(img_rgb)
    plt.axis('off')
    plt.show()