AllentDan / LibtorchDetection

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

模型训练需要有一个预训练模型yolo4_tiny.pt,这一个是否可以不要? #2

Closed chenshida closed 2 years ago

chenshida commented 2 years ago

你好,用您的代码在跑训练函数时,需要有一个预训练模型,这个预训练模型可以不用吗? 如果我用新的数据集进去训练,标签和原先的不一样,那么这个预训练模型是否还是可用的? image

AllentDan commented 2 years ago

可以继续用

chenshida commented 2 years ago

感谢,测试过了,可以使用