Transfer learning: phân loại ảnh của riêng bạn với vài trăm ảnh và mô hình tiền huấn luyện
Cách dân chuyên nghiệp làm phân loại ảnh: lấy mô hình đã học trên ImageNet (torchvision, timm), linear probing rồi fine-tuning toàn bộ với learning rate theo tầng; chuẩn bị dữ liệu bằng ImageFolder, augmentation, xử lý lệch lớp, phân tích lỗi và chọn backbone.
9 phút đọc
Bài trước, ta huấn luyện CNN từ đầu với 50.000 ảnh. Trong dự án thật, bạn thường chỉ có vài trăm ảnh — ảnh sản phẩm lỗi trên dây chuyền, ảnh lá cây bệnh, ảnh hoá đơn. Huấn luyện từ đầu với ngần đó dữ liệu sẽ overfit ngay lập tức.
Transfer learning giải quyết bằng cách mượn một mô hình đã học trên hàng triệu ảnh (ImageNet, hoặc lớn hơn), giữ lại phần "biết nhìn", chỉ dạy nó phân biệt các lớp của bạn. Đây là cách phần lớn bài toán phân loại ảnh được giải trong thực tế.
Bài viết PRO
Phần còn lại dành cho thành viên PRO
Bài này thế nào?
Bài viết có hữu ích với bạn?
Đánh giá giúp mình biết nên viết thêm về chủ đề nào.