exmee
Results
2
issues of
exmee
使用子项目rsdet成功训练出ckpt模型之后，我需要把模型转换成pb格式，查看代码发现输出节点应该是在命名“postprocess_detctions”中，但我使用tensorboard查看计算图graph时，没有发现该输出节点，打印ckpts所有节点也没有发现该节点。请问我如何才能找到模型的输出节点名称output_node_names，并且生成pb格式的权重呢？ 感谢，期待您的回复！
原版的model_0.88_deploy.onnx权重的输出是四个维度的，shape为[1, 2, h, w]，但是您网盘上的权重输出只有两个维度，shape为[h, w]，看模型结构发现您的网络除了去掉了BN层之外，在网络的最后还加了reshape、gather、reshape三个操作改变了输出维度。您可以给出修改之后的网络代码或者这三个操作的参数和实现方式吗？ 期待您的回复！