How to train custom yolo model for Maix-II v831?

sipeed / maix_train

k210(MaixPy)/V831 model example train code, include mobilenet classifier and YOLO V2 detector

https://wiki.sipeed.com/maixpy

Apache License 2.0

83 stars 40 forks source link

How to train custom yolo model for Maix-II v831? #19

Open Q2Learn opened 2 years ago

Q2Learn commented 2 years ago

Hi,

What is the latest resource on setting up and training a custom yolo model object detector for the maix-II v831 development board?

I am to understand that it needs to be done in PyTorch with https://github.com/sipeed/maix_train/tree/master/pytorch repository but is there any updated tutorial on how to set this up for training and moving the completed model to the board?

Thanks for any help on this issue.