使用TensorFlow.js转换器转换frozen model报unrecognized arguments错误
TensorFlow.js frozen模型转换unrecognized arguments报错解决
错误原因
报错由命令参数格式不符合规范导致,存在两个语法错误:
--output_node_names参数与取值之间的等号两侧不允许加空格,命令行解析器会将空格后的内容识别为独立的未知参数,触发报错- 多个输出节点名之间不要添加多余空格,避免参数解析异常
修复后的转换命令
tensorflowjs_converter --input_format=tf_frozen_model --output_node_names="detection_boxes,detection_scores,detection_classes,num_detections" ./models tfjsmodel
额外注意事项
- 执行命令前请确认当前工作目录下存在
./models文件夹,且文件夹内包含合法的frozen model pb文件 - 若使用新版本tensorflowjs,可添加
--quantize_uint8参数对输出模型进行量化压缩,大幅减小模型体积
内容的提问来源于stack exchange,提问作者Nimesh Kumar
相关产品推荐
相关产品推荐

