私信  •  关注

Romzie

Romzie 最近回复了
5 年前
回复了 Romzie 创建的主题 » mobilenetv2 tflite不需要python3的输出大小

在重读 export_tflite_ssd_graph.py 脚本中,似乎有一个选项来设置保持的最大检测次数。

把这个设为100解决了我的问题。我感觉不好。

python3 object_detection/export_tflite_ssd_graph.py \                                            
    --pipeline_config_path=$model_dir/pipeline.config \                                          
    --trained_checkpoint_prefix=$model_dir/model.ckpt \                                          
    --output_directory=$output_dir \                                                             
    --add_post_processing_op=true

python3 object_detection/export_tflite_ssd_graph.py \                                            
    --pipeline_config_path=$model_dir/pipeline.config \                                          
    --trained_checkpoint_prefix=$model_dir/model.ckpt \                                          
    --output_directory=$output_dir \                                                             
    --add_post_processing_op=true \                                                              
    --max_detections=100