MCUNet for detection on custom dataset

[Chadlikouider]

Greetings, First of all, your research is truly fascinating and groundbreaking, I enjoyed reading your papers and I'm looking forward to your future works. I would like to use your MCUNet model for human detection from a UAV camera. However, I couldn't figure out find how to train the model using a custom dataset. as well as how to import the custom dataset (Images+annotation) for training. I sincerely hope you can enlighten me on how to achieve that.

Thank you

[yogawicaksana]

Hi, I would like to use the MCUNet model for another dataset too. Have you solved this problem?

Thanks in advance.

[Chadlikouider]

@yogawicaksana Sorry Mate, still stuck trying to figure it out, I think I have to write the whole code for training and detection from scratch.

[tonyzhangnxp]

@Chadlikouider , hello threre , i have a question about the training the model: the body net is pretrained by coco or training with predict header on the custom ds at the same time?