Closed muuda closed 4 months ago
下载好 训练好的模型后,使用的命令如下:
模型评估 cd ${PaddleVideo_root} python main.py --test -c pptsm_fight_frames_dense.yaml \ -w ppTSM_fight_best.pdparams
cd ${PaddleVideo_root} python main.py --test -c pptsm_fight_frames_dense.yaml \ -w ppTSM_fight_best.pdparams
模型导出 cd ${PaddleVideo_root} python tools/export_model.py -c pptsm_fight_frames_dense.yaml \ -p ppTSM_fight_best.pdparams \ -o inference/ppTSM
cd ${PaddleVideo_root} python tools/export_model.py -c pptsm_fight_frames_dense.yaml \ -p ppTSM_fight_best.pdparams \ -o inference/ppTSM
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?
下载好 训练好的模型后,使用的命令如下:
模型评估
cd ${PaddleVideo_root} python main.py --test -c pptsm_fight_frames_dense.yaml \ -w ppTSM_fight_best.pdparams
模型导出
cd ${PaddleVideo_root} python tools/export_model.py -c pptsm_fight_frames_dense.yaml \ -p ppTSM_fight_best.pdparams \ -o inference/ppTSM
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?