FastDeploy icon indicating copy to clipboard operation
FastDeploy copied to clipboard

Yolov5seg预处理固定为640问题

Open panp4n opened this issue 2 years ago • 4 comments


温馨提示:根据社区不完全统计,按照模板提问,可以加快回复和解决问题的速度


环境

  • 【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 感谢大佬

panp4n avatar Aug 23 '23 07:08 panp4n