Yolov8 Image Segmentation 사용

Cloud_ Ghost·2023년 7월 24일
0

AI

목록 보기
8/37

https://www.youtube.com/watch?v=75LI9MI9eEo 의 내용을 옮겼습니다.

가장 먼저 anaconda prompt를 켠 다음

conda create -n yolov8 python=3.9

를 입력합니다.

다음으로

conda activate yolov8

을 입력하여 가상환경을 만들어줍니다.

(base)에서 (yolov8)으로 바뀐것을 확인할 수 있습니다.

이후

pip install ultralytics

를 입력하여 install을 진행합니다.

다음으로 python을 입력하여 python환경으로 변경 후

import torch
torch.cuda.is_available()

을 입력합니다.
결과로 False가 나올텐데 상관 없습니다.
다음 입력으로

torch.__version__

을 통해서 torch을 version을 확인합니다.

완료 후 https://pytorch.org/get-started/locally/ 에서 나오는 화면을 통해 각자 환경에 맞는 version을 클릭하여 작성할 코드 내용을 찾습니다.


저는 2.0.1에 window환경, pip package를 사용하고 language는 python
cuda11.7의 platform을 사용했습니다.

아마 이 부분이 유튜브 영상과 가장 다를 것이라고 생각드는데 영상에서는 1.13.1 version으로 나오기 때문에 당황하지 마시고 torch.version 을 입력했을 때 나오는 version을 사용하시면 됩니다.

pip3 install --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

아래에 나오는 Run this Command를 복사하여 anaconda prompt에 붙여넣기 합니다.

이 때 install의 뒤에 --upgrade를 넣어주셔야 됩니다!

(이게 좀 시간이 걸리니까 다운 하는 시간 동안 다들 좀 쉬시다가 다시 하시면 될 거 같습니다. 저도 다운하는데 시간이 걸려서 잠깐 밖에 나갔다 왔습니당 정확히 기억은 안 나는데 한 30분 넘게 걸린거 같네요...!)


다시 python을 입력해서 python환경으로 변경 후

import torch
torch.cuda.is_available()

을 입력하여 결과가 True가 나오면 됩니다.

그 후 exit() 을 입력하여 벗어납니다.


다음 user에 yolov8 폴더를 새로 생성합니다.

폴더 안에 사용할 이미지를 저장합니다.

저는 저작권 없는 이미지 중 아무거나 사용했습니다.
저장 하셨다면 다시 anaconda prompt를 켜고

cd yolov8
yolo task=detect mode=predict model=

까지 입력 후 https://github.com/ultralytics/ultralytics 에 들어가서
Detection을 확인 후 사용하실 모델을 뒷 부분에 붙여주시면 됩니다.
저는 영상과 같이 YOLOv8n을 사용하겠습니다.

yolo task=detect mode=predict model=yolov8n.pt source=people.jpg

여기서 source=~.jpg 같은 경우 본인이 yolov8 폴더에 저장했던 이미지의 저장명을 작성해주시면 됩니다.
저는 people.jpg로 사진을 저장해서 이렇게 작성했습니다.
가령 이미지 저장을 gogogo.jpg로 했다면 source=gogogo.jpg가 되는 겁니당.


하고 나면 마지막줄에 보시면 Results saved to runs\detect\predict 라고 보이실텐데
다시 아까 사진을 저장했던 yolov8 폴더에 가보면 새롭게 run이란 폴더와 yolov8n.pt가 생긴 것을 확인하실 수 있을 겁니다.

그럼 runs > detect > predict 순으로 폴더를 들어가면 결과물이 나와있을겁니다.

이렇게 사람을 검출한 이미지를 확인하실 수 있습니다.
하지만 정확히 사람 얼굴이 나온 이미지만 확인하고 싶으시다면

yolo task=detect mode=predict model=yolov8n.pt source=people.jpg conf=0.5

를 입력하게 된다면 predict2 라는 폴더에서 새로운 결과를 확인하실 수 있습니다.

runs>detect>predict2 순으로 들어가서 나온 결과 이미지를 확인해보시면

처음의 결과와 달라진 것을 확인할 수 있습니다.

profile
도망쳐서 도착한 곳에 낙원이란 있을 수 없는 거야.

0개의 댓글