Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

pretrain model #7

Open
hankhaohao opened this issue Mar 29, 2023 · 1 comment
Open

pretrain model #7

hankhaohao opened this issue Mar 29, 2023 · 1 comment

Comments

@hankhaohao
Copy link

你好,请问你的这个yolov4_tiny.pt这个权重是怎么得到的。我如果用libtorch 搭了一个yolov4,我该如何获取我的yolov4的预训练权重,谢谢

@AllentDan
Copy link
Owner

你好,请问你的这个yolov4_tiny.pt这个权重是怎么得到的。我如果用libtorch 搭了一个yolov4,我该如何获取我的yolov4的预训练权重,谢谢

可以自己从头开始用 LibTorch 训练一个。也可以找一个 Yolov4 的 PyTorch 训练的模型导出到 TorchScript。但是要保证这个 PyTorch 模型结构和 LibTorch 模型结构一致,包括权重的命名

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants