Tô màu cho ảnh xám dùng mạng nơ-ron tích chập

Tóm tắt

Tô màu cho ảnh xám có ý nghĩa quan trọng trong việc phục hồi ảnh cũ và làm cho ảnh sinh động hơn. Gần đây với sự thành công của mạng nơ-ron nhiều lớp trong lĩnh vực thị giác máy tính, mạng nơ-ron tích chập nhiều lớp cũng mang lại những kết quả tích cực đối với bài toán tô màu. Tuy nhiên, các nghiên cứu trước đây thường tập trung vào một tập dữ liệu nào đó và ít quan tâm đến sự tác động của các bộ dữ liệu khác nhau đến kết quả thu được. Bài báo này sẽ xây dựng và đánh giá các mô hình tô màu khác nhau trên nhiều tập dữ liệu, dựa trên các tiêu chí về chất lượng ảnh. Kết quả thực nghiệm thể hiện sự hiệu quả của mô hình mạng residual trong bài toán tô màu. Đồng thời, bài báo còn cho thấy việc lựa chọn tập dữ liệu huấn luyện phù hợp sẽ góp phần xây dựng một mô hình tô màu hiệu quả.