PaddlePaddle / PaddleVideo

Awesome video understanding toolkits based on PaddlePaddle. It supports video data annotation tools, lightweight RGB and skeleton based action recognition model, practical applications for video tagging and sport action detection.
Apache License 2.0
1.51k stars 376 forks source link

打架识别模型 转onnx后,onnx模型输入的第一维(batch size)为动态维度,如何将其设置为1? #695

Closed muuda closed 4 months ago

muuda commented 4 months ago

下载好 训练好的模型后,使用的命令如下:

  1. 模型评估 cd ${PaddleVideo_root} python main.py --test -c pptsm_fight_frames_dense.yaml \ -w ppTSM_fight_best.pdparams

  2. 模型导出 cd ${PaddleVideo_root} python tools/export_model.py -c pptsm_fight_frames_dense.yaml \ -p ppTSM_fight_best.pdparams \ -o inference/ppTSM

    1. 转onnx:paddle2onnx --model_dir=./inference/ppTSM --model_filename=ppTSMv2.pdmodel --params_filename=ppTSMv2.pdiparams --save_file=./weights/ucf101/ppTSMv2/pptsmv2.onnx --opset_version=11 --enable_onnx_checker=True

    但是打架识别模型 转onnx后,onnx模型输入的第一维(batch size)为动态维度,如何将其设置为1?