Kears/TensorFlow 层UpSampling2D()无法部署到 TensorRT（已知行为）。
我试图通过用UpSampling2D()其他具有并行行为的 Keras 层替换该层来找到解决方案。
理论上Conv2DTranspose()应该通过设置特定的权重并在训练中固定层的权重来完成这项工作。
我正在寻找有关如何做到这一点的帮助。
问问题
290 次
1 回答
0
我通过将模型中的所有 UpSampling 2D() 替换为 Conv2DTranspose() 进行了测试运行，然后将其转换为 UFF。（我只对模型进行了 1 个 epoch 的训练以节省时间）。然后转换器抱怨 DataFormatVecPermute。
Converting conv2d_transpose_1/conv2d_transpose-0-VecPermuteNHWCToNCHW-LayoutOptimizer as custom op: DataFormatVecPermute
Warning: No conversion function registered for layer: DataFormatVecPermute yet.
C++ 中的解析器也无法成功解析该模型。
于 2019-07-18T16:23:39.990 回答