mask-rcnn-keras
mask-rcnn-keras copied to clipboard
bubbliiiing
有人试过转换成tflite吗？
我在打算将训练出来的模型转换成tflite格式在移动端上使用，但是转换的时候一直报错。
ValueError: Unknown layer: ProposalLayer
类似这些自定义层的问题，我都在加载model的时候都添加了这个参数解决
custom_objects={"ProposalLayer": ProposalLayer}
然后将自定义层的初始化方法的参数都加上了默认的值，最终还会报错
AttributeError: 'Node' object has no attribute 'output_masks'
ValueError: Please initialize TimeDistributed layer with a Layer instance. You passed: <mrcnn.model.BatchNorm object at 0x7efc20fdf828>
我也在尝试，也加了custom 但是遇到这个问题
解决了，同时遇到了1楼的问题也解决了： https://blog.csdn.net/qq_30638831/article/details/84193517
最后转成tflite成功，但是加载的时候报错： RuntimeError: tensorflow/lite/kernels/strided_slice.cc:152 op_context.input->type != op_context.output->type (2 != 6)Node number 243 (STRIDED_SLICE) failed to prepare.
基本搜不到相关资料...