当我尝试将 faster_rcnn_inception_v2_coco 保存的模型转换为 Javascript 格式时,我收到了一个问题:
KeyError: <tf.Tensor 'SecondStagePostprocessor/BatchMultiClassNonMaxSuppresion/map/while/Switch_1:0' shape=<unkknown> dtype=float32>
环境:python3.6,tensorflowjs 1.0.1,MacOS Mojave 10.14.3
重现步骤:
- 下载faster_rcnn_inception_v2_coco tar并解压
- 尝试转换
tensorflowjs_converter --input_format=tf_saved_model --output_format=tfjs_graph_model <pathToExtractedTar>/saved_model ./output_folder
您能根据这个问题提出一些想法吗?
谢谢