diff --git a/mindspore/lite/test/models_caffe.cfg b/mindspore/lite/test/models_caffe.cfg
index bc716907e6..d414e9b2aa 100644
--- a/mindspore/lite/test/models_caffe.cfg
+++ b/mindspore/lite/test/models_caffe.cfg
@@ -69,3 +69,5 @@ ml_tabel_recog
 ml_text_division
 6c_seg_nomean_20200610
 ml_video_edit_person_divison
+ml_video_edit_img_segment
+ml_video_edit_video_segment_gauss_adaptis_part1
diff --git a/mindspore/lite/test/models_with_several_inputs_or_without_outputs.cfg b/mindspore/lite/test/models_with_several_inputs_or_without_outputs.cfg
index 24ca2212b7..feb3d6dfcd 100644
--- a/mindspore/lite/test/models_with_several_inputs_or_without_outputs.cfg
+++ b/mindspore/lite/test/models_with_several_inputs_or_without_outputs.cfg
@@ -31,3 +31,7 @@ quant_aware_bank_card_detection_inception.onnx
 quant_aware_bank_card_recognition_fcny.onnx
 quant_aware_identify_card_detect.onnx
 porseg_tmp.onnx;2
+ml_video_edit_img_segment_adaptise.pb;2
+ml_video_edit_img_segment_adaptise_pb2tflite.tflite;2
+ml_video_edit_video_segment_gauss_adaptis_part2.pb;2
+ml_video_edit_video_segment_gauss_adaptis_part2_pb2tflite.tflite;2
diff --git a/mindspore/lite/test/run_benchmark_nets.sh b/mindspore/lite/test/run_benchmark_nets.sh
index bf3dff47e3..1bcc2a8be2 100755
--- a/mindspore/lite/test/run_benchmark_nets.sh
+++ b/mindspore/lite/test/run_benchmark_nets.sh
@@ -283,6 +283,9 @@ function Run_Converter() {
         model_name=${line%%;*}
         model_type=${model_name##*.}
         case $model_type in
+          pb)
+            model_fmk="TF"
+            ;;
           tflite)
             model_fmk="TFLITE"
             ;;