Các công cụ chuẩn bị dữ liệu

Một phần của tài liệu Xây dựng hệ thống tìm kiếm ảnh theo nội dung 9 (Trang 34 - 38)

3.2.1.1. Công cụ chuẩn bị dữ liệu xác định vị trí khuôn mặt thủ công làm dữ liệu huấn luyện cho bộ phân lớp xác định vị trí mặt ngƣời

Đây là một chƣơng trình python đơn giản để chuẩn bị dữ liệu thủ công. Các bƣớc chuẩn bị dữ liệu nhƣ sau:

23

Băt đầu

Load thư mục chứa các file cần gán nhãn

Kiểm tra thư mục có file ảnh không Hiển thị ảnh Người dùng khoanh vùng vị trị khuôn mặt và hai mắt Cắt ảnh theo vùng đã khoanh, chuyển sang màu đơn sắc, xoay ảnh sao cho 2 mắt song song với

mép ảnh

Chuyển ảnh đến thư mục dữ liệu huấn

luyện gán nhãn

Kết thúc

24

Giao diện của công cụ chuẩn bị dữ liệu:

Hình 3.2: Giao diện của công cụ chuẩn bị dữ liệu huấn luyện xác định vị trí khuôn mặt

3.2.1.2. Công cụ huấn luyện bộ phân lớp xác định vị trí khuôn mặt

Đây là một chƣơng trình cài đặt theo thuật toán bootstrapping đã nêu ở trên. Sau đây là các bƣớc thực thi của chƣơng trình.

25 Huấn luyện Bộ phân

lớp M xác định vị trí khuôn mặt dựa vào bộ ảnh A1 và bộ ảnh

C

Kiểm tra độ chính xác của mô hình phân lớp với bộ ảnh test A2 đã hội

tụ chưa Kết thúc Đã hội tụ Sử dụng bộ phân lớp M để xác định vị trí khuôn mặt trên bộ ảnh B Chuyển các ảnh đã gán nhãn và xử lý của bộ ảnh B sang thư mục bộ ảnh C Bắt đầu Chưa hội tụ Load thư mục chưa

bộ ảnh đã gán nhãn A1

Load thư mục chứa bộ ảnh mở rộng C

Cắt ảnh theo vùng đã khoanh, chuyển sang màu đơn sắc, xoay ảnh sao cho 2 mắt song song với

mép ảnh

Hình 3.3: Các bƣớc thực thi của công cụ huấn luyện mô hình phân lớp xác định vị trí khuôn mặt

26

Một phần của tài liệu Xây dựng hệ thống tìm kiếm ảnh theo nội dung 9 (Trang 34 - 38)