Hỏi về cách sử dụng toán tử tf.image.non_max_suppression trong Tensorflow Lite


#1

Chào các anh chị,

Em hiện đang tìm hiểu về Tensorflow Lite và có thực hiện chuyển đổi một số mô hình về xử lý ảnh từ Tensorflow (TF) sang Tensorflow Lite (TF Lite) tuy nhiên gặp vấn đề như sau: Bên TF Lite không hỗ trợ toán tử tf.image.non_max_suppression (em không muốn để tùy chọn cho phép TF Lite sử dụng toán tử của TF vì vấn đề kích thước của ứng dụng). Có anh chị nào có cách giải quyết cho toán tử này không ạ?

Em cảm ơn nhiều.


#2

Tự viết hàm thôi em :smiley:


#3

Vâng ạ. Cơ bản các toán tử trong Tensorflow nó được viết bằng C++ xong biên dịch sẵn nên tốc độ thực thi nhanh, mình cài đặt lại nó lâu hơn, trong khi đó thời gian NMS nó khéo bằng cả thời gian chạy model CNN rồi. Tuy vậy, em cũng xem hóng được cách người ta sử dụng như thế nào rồi ạ. Cảm ơn anh!