Wang Xinyu
Wang Xinyu
TRT模型在不同显卡不通用
Thanks for finding this. In yolov5, it was using `./coco_calib/` https://github.com/wang-xinyu/tensorrtx/blob/c889b84df2e081d870bc28680f674b47070cf1d6/yolov5/src/model.cpp#L356 Can you help raise a PR?
Refer this https://github.com/wang-xinyu/tensorrtx/blob/master/tutorials/contribution.md
需要改代码，基于v5吧
You need to modify the draw_bbox function https://github.com/wang-xinyu/tensorrtx/blob/47baebce6089b5a6ae1d6bacb7589bee8f619cd7/yolov8/src/postprocess.cpp#L120
Did you re-serialize the model?
欢迎提pr
@rlewkowicz We recently added the pre-commit check, can you please follow https://github.com/wang-xinyu/tensorrtx/blob/master/tutorials/contribution.md to run pre-commit on this changed file? ``` pre-commit run --files yolov9/plugin/yololayer.h ```
在https://github.com/ultralytics/ultralytics/tree/main 的主目录下执行gen_wts.py
试试640x640输入