Darknet: Train YOLO using own data set

Created on 23 Feb 2018  Â·  6Comments  Â·  Source: pjreddie/darknet

I want retrain the yolo for only detect single class as a exmaple only want detect "STOP" sign .So i want remove all other weight files and I want only my weight file how can i do this.

Most helpful comment

change the yolo-voc.cfg file ,batches=64,subdivision=8, last convolution layer's filter=30 and classes in region layer to be 1 and modify your_data.data and your_data.names files . then start training
./darknet detector train cfg/your_data.data cfg/yolo-voc.cfg darknet19_448.conv.23
Then use this trained model for one class detector

All 6 comments

change the yolo-voc.cfg file ,batches=64,subdivision=8, last convolution layer's filter=30 and classes in region layer to be 1 and modify your_data.data and your_data.names files . then start training
./darknet detector train cfg/your_data.data cfg/yolo-voc.cfg darknet19_448.conv.23
Then use this trained model for one class detector

Can you explain bit more IM very new to YOLO

I made a document to help me and my colleagues remember how to do this. Maybe it could help you too: https://github.com/drforester/Guide_to_using_YOLOv2

Thank you !!

On Tue, Feb 27, 2018 at 8:15 AM, David Forester notifications@github.com
wrote:

I made a document to help me and my colleagues remember how to do this.
Maybe it could help you too: https://github.com/drforester/
Guide_to_using_YOLOv2

—
You are receiving this because you authored the thread.
Reply to this email directly, view it on GitHub
https://github.com/pjreddie/darknet/issues/487#issuecomment-368728088,
or mute the thread
https://github.com/notifications/unsubscribe-auth/Aav-OTzoKI7_auIdlVwV8sLNj79xIOaKks5tY2xhgaJpZM4SQrXg
.

@rucsacman did you fork the repository containing the guide? drforester account has been deleted

It was archived: https://web.archive.org/web/20180617164046/https://github.com/drforester/Guide_to_using_YOLOv2

Was this page helpful?
0 / 5 - 0 ratings