Ultralytics YOLOv8 ๐์ ์ฌ์ฉํ ๊ฑฐ๋ฆฌ ๊ณ์ฐ
๊ฑฐ๋ฆฌ ๊ณ์ฐ์ด๋ ๋ฌด์์ธ๊ฐ์?
๋ ๋ฌผ์ฒด ์ฌ์ด์ ๊ฐ๊ฒฉ์ ์ธก์ ํ๋ ๊ฒ์ ์ง์ ๋ ๊ณต๊ฐ ๋ด์์ ๊ฑฐ๋ฆฌ ๊ณ์ฐ์ด๋ผ๊ณ ํฉ๋๋ค. ์ ๊ฒฝ์ฐ Ultralytics YOLOv8์ ๊ฒฝ์ฐ ์ฌ์ฉ์๊ฐ ๊ฐ์กฐ ํ์ํ ๊ฒฝ๊ณ ์์์ ๊ฑฐ๋ฆฌ๋ฅผ ๊ณ์ฐํ๊ธฐ ์ํด ๊ฒฝ๊ณ ์์ ์ค์ฌ์ด ์ฌ์ฉ๋ฉ๋๋ค.
Watch: ๋ค์์ ์ฌ์ฉํ์ฌ ๊ฑฐ๋ฆฌ ๊ณ์ฐ Ultralytics YOLOv8
๋น์ฃผ์ผ
๋ค์์ ์ฌ์ฉํ์ฌ ๊ฑฐ๋ฆฌ ๊ณ์ฐ Ultralytics YOLOv8 |
---|
๊ฑฐ๋ฆฌ ๊ณ์ฐ์ ์ฅ์ ์?
- ํ์งํ ์ ๋ฐ๋: ์ปดํจํฐ ๋น์ ์์ ์์ ์ ํํ ๊ณต๊ฐ ์์น ํ์ ์ ํฅ์์ํต๋๋ค.
- ํฌ๊ธฐ ์ถ์ : ๋ ๋์ ์ปจํ ์คํธ ์ดํด๋ฅผ ์ํด ๋ฌผ๋ฆฌ์ ํฌ๊ธฐ๋ฅผ ์ถ์ ํ ์ ์์ต๋๋ค.
- ์ฅ๋ฉด ์ดํด: ํ๊ฒฝ์ ์ ์ฒด์ ์ผ๋ก ์ดํดํ์ฌ ์์ฌ ๊ฒฐ์ ์ ๊ฐ์ ํ๋ ๋ฐ ๊ธฐ์ฌํฉ๋๋ค.
๊ฑฐ๋ฆฌ ๊ณ์ฐ
- ๋ง์ฐ์ค ์ผ์ชฝ ๋ฒํผ์ผ๋ก ๋ ๊ฐ์ ๊ฒฝ๊ณ ์์๋ฅผ ํด๋ฆญํ์ฌ ๊ฑฐ๋ฆฌ๋ฅผ ๊ณ์ฐํฉ๋๋ค.
YOLOv8 ์ ์ฌ์ฉํ ๊ฑฐ๋ฆฌ ๊ณ์ฐ ์์
from ultralytics import YOLO
from ultralytics.solutions import distance_calculation
import cv2
model = YOLO("yolov8n.pt")
names = model.model.names
cap = cv2.VideoCapture("path/to/video/file.mp4")
assert cap.isOpened(), "Error reading video file"
w, h, fps = (int(cap.get(x)) for x in (cv2.CAP_PROP_FRAME_WIDTH, cv2.CAP_PROP_FRAME_HEIGHT, cv2.CAP_PROP_FPS))
# Video writer
video_writer = cv2.VideoWriter("distance_calculation.avi",
cv2.VideoWriter_fourcc(*'mp4v'),
fps,
(w, h))
# Init distance-calculation obj
dist_obj = distance_calculation.DistanceCalculation()
dist_obj.set_args(names=names, view_img=True)
while cap.isOpened():
success, im0 = cap.read()
if not success:
print("Video frame is empty or video processing has been successfully completed.")
break
tracks = model.track(im0, persist=True, show=False)
im0 = dist_obj.start_process(im0, tracks)
video_writer.write(im0)
cap.release()
video_writer.release()
cv2.destroyAllWindows()
์ฐธ๊ณ
- ๋ง์ฐ์ค ์ค๋ฅธ์ชฝ ํด๋ฆญํ๋ฉด ๊ทธ๋ ค์ง ๋ชจ๋ ์ ์ด ์ญ์ ๋ฉ๋๋ค.
- ๋ง์ฐ์ค ์ผ์ชฝ ํด๋ฆญ์ผ๋ก ์ ์ ๊ทธ๋ฆด ์ ์์ต๋๋ค.
์ ํ์ ์ธ์ set_args
์ด๋ฆ | ์ ํ | ๊ธฐ๋ณธ๊ฐ | ์ค๋ช |
---|---|---|---|
names |
dict |
None |
ํด๋์ค ์ด๋ฆ |
view_img |
bool |
False |
์นด์ดํธ๊ฐ ์๋ ํ๋ ์ ํ์ |
line_thickness |
int |
2 |
๋ฐ์ด๋ฉ ์์ ๋๊ป ๋๋ฆฌ๊ธฐ |
line_color |
RGB |
(255, 255, 0) |
๋ ๊ฐ์ ๊ฒฝ๊ณ ์์์ ๋งคํ๋๋ ์ค์ฌ์ฒด์ ์ ์์ |
centroid_color |
RGB |
(255, 0, 255) |
๊ฐ ๊ฒฝ๊ณ ์์์ ์ค์ฌ ์์ |
์ธ์ model.track
์ด๋ฆ | ์ ํ | ๊ธฐ๋ณธ๊ฐ | ์ค๋ช |
---|---|---|---|
source |
im0 |
None |
์ด๋ฏธ์ง ๋๋ ๋น๋์ค์ ์์ค ๋๋ ํ ๋ฆฌ |
persist |
bool |
False |
ํ๋ ์ ๊ฐ ํธ๋ ์ง์ |
tracker |
str |
botsort.yaml |
์ถ์ ๋ฐฉ๋ฒ '๋ฐ์ดํธํธ๋' ๋๋ '๋ด์ํธ' |
conf |
float |
0.3 |
์ ๋ขฐ ์๊ณ๊ฐ |
iou |
float |
0.5 |
IOU ์๊ณ๊ฐ |
classes |
list |
None |
ํด๋์ค๋ณ๋ก ๊ฒฐ๊ณผ๋ฅผ ํํฐ๋งํฉ๋๋ค(์: classes=0 ๋๋ classes=[0,2,3]). |
verbose |
bool |
True |
๊ฐ์ฒด ์ถ์ ๊ฒฐ๊ณผ ํ์ |
์์ฑ 2024-01-05, ์ ๋ฐ์ดํธ 2024-03-01
์์ฑ์: ๋ฆฌ์ฆ์ ๋ฌด๋์๋ฅด (5), ๊ธ๋ ์กฐ์ฒ (2), ์์ ์ ์์ (1)