search

WongKinYiu / PartialResidualNetworks

partial residual networks
100 stars 24 forks source link

How to train own mode on EfficientNet #3

Closed tomriddle54 closed 5 years ago

tomriddle54 commented 5 years ago

Hello @WongKinYiu I want to train custom on EfficientNet any doc would be helpful

WongKinYiu commented 5 years ago

first, using "darknet.exe partial enet-coco.cfg.txt enet-coco.weights enet-coco.conv.131 131" to get coco pre-trained weight of EfficientNet backbone.

Then, Following https://github.com/AlexeyAB/darknet#how-to-train-to-detect-your-custom-objects to modify your custom cfg file.

Note that you should re-calculate anchors of your custom dataset.