Cách lấy định dạng trọng số từ mô hình TenspFlow .pb?

Question

Tôi muốn tổ chức lại các nút của mô hình .pb tensorflow, vì vậy đầu tiên tôi nhận được NodeDef từ GraphDef, và lấy attr sử dụng NodeDef.attr() cho nút của "Conv2D". Tôi có thể nhận được các thông số như strides, padding, data_format, use_cudnn_on_gpu từ attr, nhưng không thể lấy thông số định dạng trọng số. Ngôn ngữ tôi sử dụng là C++. Cách tải xuống! Cảm ơn bạn!

Answer 1

Conv2D có hai yếu tố đầu vào: dữ liệu đầu tiên là dữ liệu và dữ liệu thứ hai là filter (hoặc trọng lượng), vì vậy bạn có thể chỉ cần kiểm tra định dạng của đầu vào thứ hai là Conv2D. Nếu bạn đang sử dụng C++, bạn có thể thử cách này:

# Assuming inputs: conv2d_node, node_map. 
filter_node_name = conv2d_node.input(1) 
filter_node = node_map[filter_node_name] 
# You might need to check identity node here. 
# Get the shape of filter_node using NodeDef.attr()