search

AllentDan / LibtorchDetection

C++ trainable detection library based on libtorch (or pytorch c++). Yolov4 tiny provided now.
MIT License
103 stars 39 forks source link

pretrain model #7

Open hankhaohao opened 1 year ago

hankhaohao commented 1 year ago

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

AllentDan commented 1 year ago

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

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