轉換官方 fused 的預訓練模型權重

bubbliiiing / yolov8-pytorch

这是一个yolov8-pytorch的仓库，可以用于训练自己的数据集。

GNU General Public License v3.0

641 stars 76 forks source link

Open summer51202 opened 1 year ago

summer51202 commented 1 year ago

B導您好，官方的預訓練模型權重有經過 fused，所以沒辦法直接 load。我 load 過您提供的 pretrained weights 發現是沒 fuse 過的，想請問您是如何做到轉換官方預訓練模型權重? 感謝!