如何利用自己的数据集训练模型？

[opentld]

trafficstars

如题。如何将自己的数据集训练检测模型并部署在jetson上？

[Qengineering]

@opentld,

You can not train the model with the dataset given in this repo. We use the ncnn framework. ncnn doesn't support training, only deploying.

Train your custom YoloV7 model on a PC. See: https://machinelearningprojects.net/train-yolov7-on-the-custom-dataset/ https://medium.com/augmented-startups/yolov7-training-on-custom-data-b86d23e6623 https://blog.paperspace.com/train-yolov7-custom-data/

Once you have your model, convert it to onnx and then to ncnn.