tensorrtx icon indicating copy to clipboard operation
tensorrtx copied to clipboard

Published 20 hours ago • verified publisher icon wang-xinyu

尝试修改输入分辨率后，pt转wts报错

Open xjcvip007 opened this issue 1 year ago • 6 comments

Env

  • GPU, RTX1060.
  • OS, Win10
  • Cuda version 11.6
  • TensorRT version TensorRT-8.4.1.5.Windows10.x86_64.cuda-11.6.cudnn8.4

About this repo

  • which branch/tag/commit are you using? master
  • which model? yolov5, retinaface? yolov8

Your problem

使用Ultralytics工程训练了一个yolov8_seg模型，分辨率为320*320，类别为7类，修改了yolov8的c++工程中的include/config.h文件，分辨率修改从640->320， 图片 重新build工程后，但是在wts转engine时会报错如下图： 图片 请问是不是model.cpp中有一些代码还是有些问题？

xjcvip007 avatar Jan 17 '24 06:01 xjcvip007

@Rex-LK 帮忙看看？

wang-xinyu avatar Jan 22 '24 03:01 wang-xinyu

@Rex-LK 帮忙看看？

好的

Rex-LK avatar Jan 23 '24 11:01 Rex-LK

图片 应该是head部分没有自适应变化，修改如上后正常了，但是320 * 320的效果比640 * 640的差很多

xjcvip007 avatar Jan 24 '24 03:01 xjcvip007

图片 应该是head部分没有自适应变化，修改如上后正常了，但是320 * 320的效果比640 * 640的差很多

麻烦你这边提个pr让 @wang-xinyu 看看吧

Rex-LK avatar Jan 24 '24 05:01 Rex-LK

嗯呢， @xjcvip007 帮忙提个PR吧

wang-xinyu avatar Jan 25 '24 03:01 wang-xinyu

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

stale[bot] avatar Apr 16 '24 05:04 stale[bot]