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.

Sign up for GitHub

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

Jump to bottom

pretrain model #7

Open

hankhaohao opened this issue Mar 29, 2023 · 1 comment

hankhaohao commented Mar 29, 2023

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

Owner

AllentDan commented Mar 30, 2023

你好，请问你的这个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