search

wang-xinyu / tensorrtx

Implementation of popular deep learning networks with TensorRT network definition API
MIT License
6.92k stars 1.77k forks source link

yolov5-7.0，about anchor #1530

Open cgreenhand opened 4 months ago

cgreenhand commented 4 months ago

Env

About this repo

Your problem

博主您好，请教一下，gen_wts.py产生的yolov5_detect 的wts文件里面包含模型的anchor信息吗，使用U版yolov5训练自己的模型，开启autoanchor后anchor就和coco的不一致了，通过wts转换engine里面会包含更新后的anchor吗？还是固定的coco anchor，如果转换是固定的coco-anchor我们又该怎么修改anchor呢？谢谢

wang-xinyu commented 4 months ago

包含anchor

cgreenhand commented 4 months ago

谢谢博主的解答

stale[bot] commented 1 month ago

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.