Teachablemachine – công cụ huấn luyện phân loại ảnh

tensorflow-js-logo-social

Teachablemachine – công cụ huấn luyện phân loại ảnh online giúp bạn phân loại ảnh – classification. Công cụ do Google cung cấp giúp bạn training model ngay trên web và test trên web.

Các bạn truy cập vào link https://teachablemachine.withgoogle.com/ để sử dụng.

Tool này cực kỳ đơn giản, dễ sử dụng đối với người không biết gì về AI, ML. Chỉ cần có bộ ảnh được phân loại là bạn có thể training, thậm chí không có ảnh bạn cũng có thể sử dụng webcam để lấy hình. Không cần đăng ký, cứ vào web https://teachablemachine.withgoogle.com/train/image là train được.

Bạn có thể sử dụng để train âm thanh và hình dáng cơ thể, trong bài này mình chỉ giới thiệu về phân loại hình ảnh. Các ứng dụng khác bạn có thể làm tương tự.

teachablemachine_application

Đây cũng là mã nguồn mở, bạn có thể clone ở https://github.com/googlecreativelab/teachablemachine-community

Nguyên lý

Khi bạn upload hình ảnh đã phân ra các class thì web sẽ tiến hành training. Sau khi training xong các bạn có thể download model để sử dụng hoặc test ngay trên web.

Teachable machine sử dụng Tensorflow.js để tiến hành traning ngay trên browser của bạn. Nghĩa là dữ liệu không hề gửi đến server của google mà sử dụng chính máy tính của bạn để training.

Thực hành

Bạn cần phải phân chia bộ ảnh ra nhiều lớp để training. Bài này mình sẽ sử dụng bộ ảnh car-lighting để training (link ở cuối bài). Bộ ảnh này có 2 lớp là xe hơi bật hoặc tắt đèn, chúng ta sẽ sử dụng để kiểm tra xe có bật hay tắt đèn. Class 1 là xe bật đèn, class 2 là xe tắt đèn. Ngoài ra bạn cũng có thể kiểm tra bằng đèn flash của điện thoại.

teachablemachine_classes

Chương trình sẽ cho bạn chỉnh các tham số để huấn luyện
teachablemachine_parameter

– Epochs: càng cao thì càng chính xác và huấn luyện càng lâu
– Batch size: hiểu đơn giản là số ảnh sẽ dùng trong mỗi step training

Bạn có thể để mặc định nếu như không hiểu sâu về các tham số.

Sau khi training xong bạn có thể test thử, lấy 1 ảnh bật đèn xe cho kết quả chính xác là class 1 => xe bật đèn.
teachablemachine_result1

Thử với 1 hình khác, kết quả là class 2 => xe tắt đèn
teachablemachine_result2

Thông tin thêm

Bạn có thể sử dụng để phân loại các ảnh như:
– Khuôn mặt nam – nữ
– Các loại bông hoa
– Vật thể bất kỳ

Download

car_lighting.zip (2.3MB – Google Drive)

Leave a Reply