Created by: mzchtx

使用下述命令进行 inference 并保存 inference model 出现 ValueError: var im_shape not in this block 错误。

python tools/infer.py -c configs/cascade_rcnn_r50_fpn_1x.yml --infer_dir=./images --save_inference_model

代码中 feeded_var_names 只删除了 im_id ，还保留了 image，im_info，im_shape ，而 CascadeRcnn 只需要 image，im_info，删了 im_shape 即可，不知道是否有些模型还会需要 im_shape？

对应的代码位置：code