FastDeploy
FastDeploy copied to clipboard
Yolov5seg预处理固定为640问题
温馨提示:根据社区不完全统计,按照模板提问,可以加快回复和解决问题的速度
环境
-
【FastDeploy版本】: 说明具体的版本,如fastdeploy-gpu-python=1.0.7
-
【编译命令】
-
【系统平台】: Windows x64(Windows10)
-
【硬件】: 说明具体硬件型号,如 Nvidia GPU 3060, CUDA 11.2 CUDNN 8.2
-
【编译语言】: python
-
【性能问题】 因项目需求需要识别细长的小目标,训练了1280*1280的Yolov5seg模型才能保证分割效果良好,但使用fastdeploy yolov5seg预测接口报错提示输入图像尺寸和模型input不一致。 查看fastdeploy yolov5seg 预处理部分发现resize尺寸是写死的640。 因为使用paddle套件训练的模型都会读取infer_cfg.yml来确认预处理的参数,所以yolov5seg能否改成根据模型input尺寸来决定预处理resize尺寸。 @wjj19950828 感谢大佬