CHƯƠNG 3 : CƠ SỞ THỰC HIỆN
3.4. Lưu đồ thuật toán
Với bài toán phát hiện vi phạm khoảng cách xã hội ta thực hiện đo khoảng cách dựa vào thuật toán phát hiện đối tượng theo các bước sau:
Bước 1: Dữ liệu đầu vào gồm ảnh hoặc video
Bước 2: Ứng dụng thuật toán yolo, sử dụng file mô hình yolo phát hiện đối tượng Bước 3: Sau khi xác định được vị trí có người, hệ thống sẽ trích xuất boundingbox của đối tượng và số điểm chính xác cho đối tượng
Bước 4: Nếu ngưỡng điểm cho đối tượng trên 0,6 thì hệ thống thực hiện đo khoảng cách giữa các tâm của boundingbox dựa vào khoảng cách Euclid, nếu ngưỡng điểm bằng hoặc dưới 0,6 kết thúc chương trình.
Bước 4: Nếu khoảng cách đo được giữa hai đối tượng lớn hơn ngưỡng cho phép là 2 mét thì hệ thống phát ra cảnh báo vi phạm khoảng cách an toàn, nếu khoảng cách đo được giữa hai đối tượng thấp hơn ngưỡng cho phép thì hệ thống kết thúc chương trình. Bước 5: Kết thúc chương trình.
Hình 3.5. Sơ đồ thuật toán đo khoảng cách dựa vào thuật toán phát hiện đối tượngĐối với bài toàn phát hiện khẩu trang, tương tự ta cũng thực hiện với các bước Đối với bài toàn phát hiện khẩu trang, tương tự ta cũng thực hiện với các bước huấn luyện và phát hiện như sau:
Bước 2: Ứng dụng thuật toán yolo vào huấn luyện mô hình với file mô hình yolo phát hiện khẩu trang bao gồm 2 lớp: lớp 0 là phát hiện người đeo khẩu trang và lớp 1 là phát hiện người không đeo khẩu trang.
Bước 3: Hệ thống trích xuất ra đối tượng và điểm chấm độ chính xác cho đối tượng. Bước 4: Nếu điểm chấm đạt trên 0.8 và nhận diện đối tượng thuộc lớp 1 thì phát ra cảnh báo về việc không đeo khẩu trang, ngược lại thì kết thúc chương trình.
Hình 3.6. Sơ đồ thuật toán phát hiện đeo khẩu trang