search

mit-han-lab / mcunet

[NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning
https://mcunet.mit.edu
MIT License
460 stars 82 forks source link

MCUNet for detection on custom dataset #17

Open Chadlikouider opened 1 year ago

Chadlikouider commented 1 year ago

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 commented 1 year ago

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

Thanks in advance.

Chadlikouider commented 1 year ago

@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 commented 1 year ago

@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?