search

kriyeng / yolo-on-colab-notebook

How to train YOLOv3 using Darknet on Colab 12GB-RAM GPU notebook and optimize the VM runtime load times
49 stars 40 forks source link

Transfer Learning #6

Open StefanPushkov opened 4 years ago

StefanPushkov commented 4 years ago

Hello! Is it possible to fit the network on my own class of objects or is it necessary to train from scratch here?

kriyeng commented 4 years ago

Hi @StefanPushkov , you can train using your own dataset and classes. It is explained here: Training yolo with your own dataset

In the following line is where you are getting a pretrained wieghts:

!./darknet detector train "/content/gdrive/My Drive/darknet/obj.data" "/content/gdrive/My Drive/darknet/yolov3.cfg" "/content/gdrive/My Drive/darknet/darknet53.conv.74" -dont_show

The weights are darknet53.conv.74

I hope this helps!