How can i train YOLO network?

Hi everyone

i crated my network to solve binary classify problem. (stone or not)

when i trained my netowork, I use photoes that just shows one stone as input data.

Then, I added yolo network to this network as transferlearing.

Doing that, when the photo which has many stones comes input on Yolo network, The results is to classify many stones with anchor boxes.

But I dont konw how to train this yolo net.

and,,, i don't know how to prepare anchor box data as yolo train data.

thank you for helping me.

Answer 1

[Vivek Akkala]

Hi,

Please refer to Object Detection Using YOLO v2 Deep Learning example that explains complete workflow which includes , preparing the dataset, training the network and evaluating the network on a test dataset.

The example also shows how to obtain anchor boxes for a given dataset.