You are on page 1of 13

Sử dụng Yolov8

cho detect và
extract object
Vấn đề cần giải quyết
• Yolo(You Only Look Once) là mô hình nhận diện các đối tượng có
trong ảnh, video.
• Do có khả năng nhận diện được đối tượng. Yolo được sử dụng để
giải quyết vấn đề detect và extract mặt trước/sau CCCD có gắn chip
này.
Đầu vào
• Mô hình sử dụng bộ dataset gồm:
• Tổng số lượng 898 ảnh có kích
thước 640x640
• Gồm các loại: CCCD có gắn chip,
CCCD không gắn chip, CMND,
khác.
• Được chia làm tất cả 206 epochs
Dữ liệu

Mặt trước CCCD Mặt sau CCCD có CCCD không có CMND


có gắn chip. gắn chip. gắn chip
Quy trình thực hiện
Output
• Độ chính xác của mô hình đạt được 99% trong mẫu train.
Output
• Kết quả các
trường hợp dự Mặt trước CCCD có Mặt sau CCCD có
đoán. gắn chip. gắn chip.

CCCD không có CMND


gắn chip
Triển khai trên Example
• Kết quả đạt được:
• Thành công nhận diện được hai CCCD
có chip.
• Độ chính xác tương đối ổn định
(87%~98%).
Đánh nhãn cho ảnh

• Sử dụng tool có sẵn: https://www.makesense.ai/ tạo nhãn cho


từng object
• Tạo project: Get start-> chọn tập dữ liệu cần đánh label-> Object
Detection -> Start project
• Xuất label: Actions-> Export-> chọn yolo format.
Đánh nhãn cho ảnh
Train
• Tải các tài nguyên phía dưới
• File train:
https://colab.research.google.com/drive/1huecpiEV2--Ituc
_4whzp-847smOybD4#scrollTo=aAiR-kYsRAKN
• Tải tập tin:
https://drive.google.com/file/d/16IsWpRHSBKuPjOeRKw8y Cây thư mục
ddbewDfsY87T/view?usp=sharing
• Tải dataset:
https://drive.google.com/drive/folders/1C2SIJURp0O4bXZ
ajZVz2qc8trUjqqEuy?usp=sharing
Lưu mô hình
• Chạy các dòng trong file code.
• Chọn định dạng muốn dùng và xuất.
• Tflite: !yolo export
model=/content/runs/detect/train2/weights/
best.pt format=tflite
• Torchscript: !yolo mode=export
model=/content/onnx-tensorflow/runs/detec
t/train/weights/best.pt format=torchscript
optimize
• Lưu mô hình và nhúng model vào app.
Reference
Thư viện: https://pub.dev/packages/pytorch_lite
TensorFlow:
https://www.tensorflow.org/lite/models/convert/metadata?hl=vi%5
C
Colab:
https://colab.research.google.com/drive/1huecpiEV2--Ituc_4whzp-8
47smOybD4#scrollTo=pR0DCvuPkXwh

You might also like