tensorflow ckpt 模型 转换 pb 模型, 测试模型是否正确, 后续实现 c++ 部署。
Code: https://github.com/QingyongHu/RandLA-Net
测试PB 模型
-
RandLANetConvert.py
import tensorflow.compat.v1 as tf tf.disable_v2_behavior
tensorflow ckpt 模型 转换 pb 模型, 测试模型是否正确, 后续实现 c++ 部署。
Code: https://github.com/QingyongHu/RandLA-Net
RandLANetConvert.py
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior