top of page
Search

การตรวจจับวัตถุโดยใช้งาน YOLOv5

Updated: Oct 20, 2023

ผมได้ศึกษาและลองเล่นเทคโนโลยีเกี่ยวกับการตรวจจับวัตถุโดยใช้ YOLOv5 ซึ่งเป็นการใช้ข้อมูลจาก Data set ที่ทางผู้พัฒนาได้จัดทำไว้ นำมาใช้ในการตรวจจับวัตถุผ่านกล้อง Webcam ของเรา

ซึ่งภาพที่จะได้ก็จะมีตัวอย่างดังภาพข้างล่าง

จากภาพเป็นการตรวจจับคนที่ได้

ขั้นตอนในการติดตั้ง ( Install )

Clone repo and install requirements.txt in a Python>=3.8.0 environment

git clone https://github.com/ultralytics/yolov5

# clonecd yolov5

pip install -r requirements.txt


ใช้ YOLOv5 ใน Python เพื่อสร้างระบบตรวจจับวัตถุแบบอัตโนมัติ

  1. ติดตั้ง YOLOv5: คุณสามารถติดตั้ง YOLOv5 ได้โดยใช้คำสั่ง pip ดังนี้: Copy code pip install yolov5

  2. รับโมเดล: คุณสามารถเลือกขนาดของโมเดล YOLOv5 ที่เหมาะกับงานของคุณ (เช่น, YOLOv5s, YOLOv5m, YOLOv5l, YOLOv5x) และดาวน์โหลดโมเดลจากเครื่องหรือเว็บไซต์ YOLOv5.

  3. เตรียมข้อมูล: คุณจะต้องเตรียมข้อมูลตรวจจับของคุณในรูปแบบที่ YOLOv5 รองรับ สามารถใช้ไฟล์ yaml เพื่อกำหนดคอนฟิกเช่นว่าจะใช้อันคอตต่อไปนี้:

yamlCopy code
train: data/train.yaml  # แทนที่ด้วยไฟล์ข้อมูลการฝึกอบval: data/val.yaml  # แทนที่ด้วยไฟล์ข้อมูลการตรวจสอบnc: 80  # จำนวนชนิดของวัตถุที่คุณต้องการตรวจจับ
  1. เริ่มการฝึกฝน: เมื่อคุณมีโมเดลและข้อมูลพร้อมแล้ว คุณสามารถเริ่มการฝึกโมเดลโดยใช้คำสั่ง:

pythonCopy code
!python train.py --img 640 --batch 16 --epochs 100 --data your_data.yaml --cfg your_model.yaml --weights your_initial_weights.pt
  1. ใช้โมเดลที่ฝึกแล้ว: เมื่อโมเดลได้รับการฝึกแล้ว คุณสามารถใช้โมเดลในการตรวจจับวัตถุโดยใช้คำสั่ง:

pythonCopy code
!python detect.py --source your_image_or_video.mp4 --weights your_trained_model.pt --conf 0.4

นี่เป็นขั้นตอนพื้นฐานในการใช้ YOLOv5 ใน Python เพื่อตรวจจับวัตถุ โดยคุณสามารถปรับแต่งคำสั่งต่าง ๆ เพื่อตรงตามความต้องการของคุณ และอ่านเอกสารและคู่มือเพิ่มเติมจากเว็บไซต์ YOLOv5 เพื่อเรียนรู้เพิ่มเติมเกี่ยวกับการใช้งานและการปรับแต่ง YOLOv5 ในโปรเจกต์ของคุณ.

14 views0 comments

Recent Posts

See All
bottom of page