search

FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎
GNU General Public License v2.0
1.25k stars 195 forks source link

yolov8-seg模型尺寸为1280问题 #129

Open ASSicat opened 1 month ago

ASSicat commented 1 month ago

我成功复现了您的代码，并输出正确结果。但是我使用自己训练的模型时能够跑通，但是最后输出的结果掩码与原图实际位置不符。我重新训练模型imgsz为640就可以正确出图，1280就不能正确出图，这是什么原因：我设置的参数如下 initParameters.dst_h = 1280; initParameters.dst_w = 1280;

m_output_seg_w = 320 * 320; m_output_seg_h = 32; 请问是还需要更改其它参数吗