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

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

Open

opentld opened this issue Sep 7, 2023 · 1 comment

opentld commented Sep 7, 2023

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

Owner

Qengineering commented Sep 7, 2023

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment