Chúng tôi đề xuất thêm với thuật toán của Swain và Ballard tại hai điểm. Thứ nhất, chúng tôi không so sánh toàn bộ lược đồ màu mà chỉ tập trung vào màu tìm kiếm. Do đó, độ đo hoàn toàn không bị gây nhiễu của các màu phụ.
Và thứ hai, chúng tôi thêm các cấp xám cần thiết vào ảnh đầu vào, do đó có thể khớp tốt hơn. Điều này nhằm san bằng sự khác biệt giữa ảnh đầu vào tổng hợp (chỉ có một cấp xám) và ảnh tự nhiên (một vùng trời có thể nhiều màu xanh da trời, một vùng cây có nhiều lá xanh non và xanh đậm).
Cách làm của chúng tôi rất đơn giản, chúng tôi định nghĩa mặt nạ so sánh và hàm so sánh dựa trên mặt nạ. Chúng tôi sẽ định nghĩa mặt nạ và chỉ so sánh những điểm trong lược đồ có mặt nạ giá trị 1. Xem Công thức 3.2.
m(k) = (hq(k) > 0); //mặt nạ m có m(k)=1 nếu như hq(k) > 0 và m(k)=0 nếu ngược lại
d(hq, hp) = sum(diff(hq(k), hp(k)) * m(k) | k=0...4095); //so sánh độ sai khác và chỉ tính những điểm nào có giá trị màu trên histogram đầu vào
Công thức 3.2: Định nghĩa mặt nạ so sánh và độ đo mới dựa trên mặt nạ. Mục đích là tập trung vào những ô nhớ có màu trên lược đồ đầu vào và bỏ qua điểm
nhiễu.
Chúng tôi nhận thấy rằng độ đo như trên là quá chặt. Nó cắt cụt tất cả các giá trị khác trong lược đồ màu tự nhiên. Ví dụ trong ảnh tự nhiên có một quả táo, nó sẽ có từ màu đỏ đậm tới màu đỏ nhạt (nhiều cấp xám). Trong khi ảnh đầu vào của chúng ta chỉ có một màu đỏ (một cấp xám). Do đó chúng ta cần tạo một cái cầu nối (bridge) giữa lược đồ đầu vào (ảnh tổng hợp vẽ bằng tay) và lược đồ
tự nhiên (ảnh chụp từ máy ảnh). Chúng tôi giới thiệu thêm một số màu sắc bằng cách sinh ra cấp xám cần thiết.
Đầu vào: ảnh tổng hợp I chỉ có một số ít cấp xám
Đầu ra: ảnh tổng hợp J có thêm một số cấp xám ngẫu nhiên
Bước 1: tạo lược đồ màu trong đó có chứa các tâm màu tại mỗi ô cùng với số lượng màu
Bước 2: với mỗi màu I(x, y) có vị trí lk, trong lược đồ, J(x, y) = C(lk) + rand Trong đó C(lk) là tâm màu của ô lk và rand là một lượng ngẫu nhiên được thêm vào
Bước 3: trả về ảnh J
Thuật toán 3.3: Thuật toán thêm cấp xám vào ảnh tổng hợp để nó có lược đồ màu gần với lược đồ màu của ảnh tự nhiên, nhằm tăng khả năng tìm thấy
(recall) của hệ thống tìm kiếm.