search

gplast / DroNet

DroNet: Efficient convolutional neural network detector for Real-Time UAV applications
62 stars 18 forks source link

training weights 250mb #3

Open joel5638 opened 3 years ago

joel5638 commented 3 years ago

hi @gplast , I'm replicating your repo for training a yolo model on custom data. My target is to get the weights less than 1MB in size. But after the training is finished I see they are around 250 MB. Did I miss anything here? or should I do another step? Can you please help me?

joel5638 commented 3 years ago

@gplast Which cfg and base weights should I use to get the required weight file less than 1MB?

I have used tiny-yolo-voc.cfg and tiny-yolo.weight. it dint work. There are no detections. im training with only 100images