Chào các tiền bối, chẵng là e đang làm một cái app về xử lý ảnh, sử dụng deep learning để biến đổi style ảnh.
Em dùng Tensorflow để training ra Checkpoint (ckpt) model, rồi freezing thành protobuff để sử dụng trên Android.
Code của e dựa vào repo này ạ:
E đang gặp phải một vấn đề lớn, đó là khi tạo graph để freeze model, e phải định batch size và shape cho input node và output node. Nhưng nếu làm như vậy thì trên android chỉ có thể truyền vào sử dụng 1 kích cở duy nhất (đúng với size đã định lúc write graph) ngoài ra nó sẽ báo lỗi không đồng nhất size.
Các tiền bối ai có kinh nghiệm cho vụ này giúp đỡ e với.
Liệu có thể tạo input node và output node mà không quy định size như sample của tensorflow đã tạo hay không ạ?