ผมได้ศึกษาและลองเล่นเทคโนโลยีเกี่ยวกับการตรวจจับวัตถุโดยใช้ 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 เพื่อสร้างระบบตรวจจับวัตถุแบบอัตโนมัติ
ติดตั้ง YOLOv5: คุณสามารถติดตั้ง YOLOv5 ได้โดยใช้คำสั่ง pip ดังนี้: Copy code pip install yolov5
รับโมเดล: คุณสามารถเลือกขนาดของโมเดล YOLOv5 ที่เหมาะกับงานของคุณ (เช่น, YOLOv5s, YOLOv5m, YOLOv5l, YOLOv5x) และดาวน์โหลดโมเดลจากเครื่องหรือเว็บไซต์ YOLOv5.
เตรียมข้อมูล: คุณจะต้องเตรียมข้อมูลตรวจจับของคุณในรูปแบบที่ YOLOv5 รองรับ สามารถใช้ไฟล์ yaml เพื่อกำหนดคอนฟิกเช่นว่าจะใช้อันคอตต่อไปนี้:
yamlCopy code
train: data/train.yaml # แทนที่ด้วยไฟล์ข้อมูลการฝึกอบval: data/val.yaml # แทนที่ด้วยไฟล์ข้อมูลการตรวจสอบnc: 80 # จำนวนชนิดของวัตถุที่คุณต้องการตรวจจับเริ่มการฝึกฝน: เมื่อคุณมีโมเดลและข้อมูลพร้อมแล้ว คุณสามารถเริ่มการฝึกโมเดลโดยใช้คำสั่ง:
pythonCopy code
!python train.py --img 640 --batch 16 --epochs 100 --data your_data.yaml --cfg your_model.yaml --weights your_initial_weights.pt
ใช้โมเดลที่ฝึกแล้ว: เมื่อโมเดลได้รับการฝึกแล้ว คุณสามารถใช้โมเดลในการตรวจจับวัตถุโดยใช้คำสั่ง:
pythonCopy code
!python detect.py --source your_image_or_video.mp4 --weights your_trained_model.pt --conf 0.4นี่เป็นขั้นตอนพื้นฐานในการใช้ YOLOv5 ใน Python เพื่อตรวจจับวัตถุ โดยคุณสามารถปรับแต่งคำสั่งต่าง ๆ เพื่อตรงตามความต้องการของคุณ และอ่านเอกสารและคู่มือเพิ่มเติมจากเว็บไซต์ YOLOv5 เพื่อเรียนรู้เพิ่มเติมเกี่ยวกับการใช้งานและการปรับแต่ง YOLOv5 ในโปรเจกต์ของคุณ.
Comments