obendidi / Tracking-with-darkflow

Real-time people Multitracker using YOLO v2 and deep_sort with tensorflow
GNU General Public License v3.0
524 stars 176 forks source link

About using .weights and .cfg files from darknet to track other objects #82

Open bit-scientist opened 5 years ago

bit-scientist commented 5 years ago

Before starting everything I thought it'd be better to verify I am doing right. My plan is: using yolo3 (by darknet provided by AlexeyAB) to train my own object for only one class. I am going to use yolov3.weights and yolov3.cfg to train my model and get the best .weights file. Then I would like to use that weights file and cfg file in these lines. If I do so, will I be able to track the class object? The class object is a safety helmet and I need to track that helmet. I am looking forward to your opinions and suggestions. Thanks