Tôi làm cách nào để thêm nhãn vào Hình ảnh TensorBoard?

Question

TensorBoard là một công cụ tuyệt vời, nhưng nó có thể mạnh mẽ hơn không? Hình ảnh dưới đây cho thấy hình ảnh trong TensorBoard.

Nó được gọi bằng đoạn mã sau:

tf.image_summary('images', images, max_images=100) 

Như API cho thấy, chữ số cuối cùng là "số hình ảnh", 0-99 trong trường hợp này vì tôi đã chỉ định max_images = 100. Tôi muốn hỏi, nếu tôi có thể gắn nhãn của hình ảnh này vào văn bản? Đây sẽ là một chức năng tuyệt vời để có được vì nó cho phép người dùng nhìn thấy trong thời gian thực các hình ảnh và nhãn tương ứng của họ trong quá trình đào tạo. Trong trường hợp một số hình ảnh hoàn toàn sai lệch, một bản sửa lỗi có thể được thực hiện. Nói cách khác, tôi muốn văn bản tương ứng trong hình ảnh bên dưới là:

images/image/9/5 
images/image/39/6 
images/image/31/0 
images/image/30/2 
where last digit is the label. 

Cảm ơn!

Answer 1

tôi đã không thể tìm thấy một cách để làm điều này chỉ sử dụng tensorflow, vì vậy thay vào đó tôi thực hiện như sau:

1. Tạo một giữ chỗ cho hình ảnh tóm tắt (ví dụ như một (10, 224, 224, 3) cho mười hình ảnh tóm tắt).
2. Tạo tóm tắt hình ảnh dựa trên trình giữ chỗ đó.
3. Trong khi xác thực (hoặc đào tạo, nếu bạn thích), hãy kéo hình ảnh và nhãn cho bản tóm tắt của bạn vào python bằng cách sử dụng một cái gì đó như session.run([sample_images, sample_labels]).
4. Lặp lại qua lô và sử dụng OpenCV để ghi nhãn lên ảnh bằng cách sử dụng cv2.putText.
5. Chạy tóm tắt op cung cấp hình ảnh được gắn nhãn cho trình giữ chỗ.