AI로 작물의 상태를 확인해보자! yolov8 (detection/segment)

1. 자료조사

1. Object Detection with YOLOv8

https://velog.io/@choonsik_mom/Object-Detection-with-yolo-NAS-zpetis4o

 

2. yolov8 custom training / YOLOv8 커스텀 학습 하는 방법

https://m.blog.naver.com/jong_jh_/222984251856

yolov8 custom training / YOLOv8 커스텀 학습 하는 방법

 

3. Ultralytics YOLOv8 - 공식 문서

https://docs.ultralytics.com/ko/models/yolov8/

YOLOv8

 

4. 설치

https://docs.ultralytics.com/ko/quickstart/#install-ultralytics

빠른 시작

 

5. github - ultralytics : YOLOV11

https://github.com/ultralytics/ultralytics

GitHub - ultralytics/ultralytics: Ultralytics YOLO11 🚀

- 이건 YOLO11 모델

 

 

 

+) mqtt를 이용한 이식

https://m.blog.naver.com/gyeongjuhs/223608825480?recommendCode=2&recommendTrackingCode=2

2. MQTT를 이용한 실시간 객체 탐지 영상 전송과 수신

 

 

 

 

2. 시작하기

- 출처 자료

https://webnautes.tistory.com/1851

---

1. 사전 설치 

pip3 install ultralytics

 

2. 소스코드

" vscode에서 실행한 코드"

import cv2
from ultralytics import YOLO

# Load the YOLOv8 model
model = YOLO('yolov8n.pt')

# 동영상 파일 사용시
# video_path = "path/to/your/video/file.mp4"
# cap = cv2.VideoCapture(video_path)

# webcam 사용시
cap = cv2.VideoCapture(0)

# Loop through the video frames
while cap.isOpened():
    # Read a frame from the video
    success, frame = cap.read()

    if success:
        # Run YOLOv8 inference on the frame
        results = model(frame)

        # Visualize the results on the frame
        annotated_frame = results[0].plot()

        # Display the annotated frame
        cv2.imshow("YOLOv8 Inference", annotated_frame)

        # Break the loop if 'q' is pressed
        if cv2.waitKey(1) & 0xFF == ord("q"):
            break
    else:
        # Break the loop if the end of the video is reached
        break

# Release the video capture object and close the display window
cap.release()
cv2.destroyAllWindows()

 

3. 실행 결과

yolo v8 테스트 이미지

 

출력

 

 

4.리뷰

출력된 창을 보면,

1 person, 1 couch, 1 bed, 1 tv가 출력되고 있음을 보인다.

현재 tv가 없기 때문에 인식을 잘못하고 있는것으로 나타난다.

 

5. 추가

추가로 모델의 종류(segment) 변경하여 테스트 하였다.

다른 모델을 사용하고 싶다면, "파일 이름"만 변경하면 된다.