Mạng kết hợp để phân loại, cực kỳ nhạy cảm với ánh sáng


8

Tôi đã đào tạo một mạng tích chập để phân loại hình ảnh của một thành phần cơ học là tốt hay khiếm khuyết. Mặc dù độ chính xác kiểm tra cao, tôi nhận ra rằng mô hình hoạt động kém trên các hình ảnh có ánh sáng hơi khác nhau.

Các tính năng mà tôi đang cố gắng phát hiện rất tinh tế và ánh sáng dường như khiến người mẫu gặp phải nhiều vấn đề tiêu cực. Hầu hết, nếu không phải tất cả, dữ liệu huấn luyện và kiểm tra đến từ những hình ảnh được chụp dưới ánh sáng khuếch tán. Những hình ảnh mới mà tôi đã thử mô hình trên được chụp với ánh sáng tập trung.

Hy vọng rằng cân bằng biểu đồ ( CLAHE ) sẽ giúp ích, tôi đã thực hiện cân bằng trên hình ảnh trước khi đưa nó vào mô hình, thực hiện điều này để đào tạo cũng như kiểm tra dữ liệu. Vấn đề sau đó là độ chính xác cao, nhưng mô hình dường như đã học được một số mối tương quan khác giữa hình ảnh và nhãn. Sau khi cân bằng, mọi thứ trong ảnh nổi bật, các khuyết điểm càng trở nên tinh tế và khó phát hiện hơn ngay cả đối với mắt người.

Dự đoán mô hình đã thay đổi ngay cả khi cùng một thành phần được chụp với cùng một ánh sáng nhưng với ánh sáng ở các hướng khác nhau. Vì vậy, câu hỏi của tôi là, tôi có thể làm gì, ở cuối dữ liệu hoặc với mô hình, để nó trở nên mạnh mẽ khi thay đổi ánh sáng (hoặc cùng loại ánh sáng nhưng thay đổi theo hướng chiếu sáng)?


Có khả thi để thu thập thêm dữ liệu với các biến thể trong ánh sáng? Sự khác biệt giữa ánh sáng khuếch tán và ánh sáng trực tiếp trong một cảnh sẽ khó mô phỏng hoặc cho phép xử lý trước.
Neil Slater

@NeilSlater Có, tôi có thể thu thập thêm dữ liệu. Tôi quan tâm nhiều hơn đến việc làm cho mô hình mạnh mẽ để thay đổi chiếu sáng, ở một mức độ hợp lý. Một điều tôi nhận thấy là, dự đoán mô hình đã thay đổi ngay cả khi cùng một thành phần được chụp dưới cùng một ánh sáng, nhưng các hướng khác nhau. Làm cho mô hình mạnh mẽ với các loại ánh sáng khác nhau có thể khó khăn, có kỹ thuật nào để làm cho nó mạnh mẽ theo các hướng khác nhau của cùng một ánh sáng không?
dpk

Trừ trung bình từ các hình ảnh thường giúp. Ngoài ra, dữ liệu của bạn được cân bằng tốt như thế nào? Nếu trong tập dữ liệu huấn luyện 1% là lớp 1, mô hình sẽ không học anytjing và phân loại tất cả là lớp 0.
Alex

@Alex Tôi chưa thử phép trừ. Cám ơn vì sự gợi ý. Ngoài ra, dữ liệu của tôi là cân bằng tốt. Tôi có 2 lớp và mỗi lớp chiếm 50% dữ liệu.
dpk

Câu trả lời:


2

Nó được gọi là quá mức, mô hình của bạn đã học cách dự đoán nhãn dựa trên các tính năng không thực sự chịu trách nhiệm cho các dự đoán cụ thể, vì vậy khi chúng không được trình bày, nó không thể dự đoán đúng. Mặc dù có nhiều cách khác nhau để đối phó với tình trạng thừa (ví dụ: Bỏ học), điều bạn dường như cần là tăng cường hình ảnh. Đó là một cách đơn giản nhưng rất mạnh mẽ để đào tạo một mạng lưới thần kinh mạnh mẽ. Đối với trường hợp của bạn - chỉ cần mô phỏng các điều kiện ánh sáng khác nhau bằng cách tăng / giảm ngẫu nhiên các giá trị pixel cho các đầu vào của bạn cho cùng một nhãn. Đó cũng là cách phổ biến để thêm tiếng ồn ngẫu nhiên, xoay ngẫu nhiên, lật, v.v.

Để biết thêm thông tin kiểm tra câu trả lời stackoverflow này .

Ngoài ra, gần đây tôi đã xuất bản dự án nơi tôi đã sử dụng các hàm Augmented khác nhau mà bạn có thể thấy hữu ích. Xem: https://github.com/Naurislv/P12.1-Semantic-Segmented/blob/master/auributionation.py

Khi sử dụng trang web của chúng tôi, bạn xác nhận rằng bạn đã đọc và hiểu Chính sách cookieChính sách bảo mật của chúng tôi.
Licensed under cc by-sa 3.0 with attribution required.