search

thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
GNU General Public License v3.0
6.13k stars 2.08k forks source link

Training on annotated video #953

Open KristianDukov opened 5 years ago

KristianDukov commented 5 years ago

Is there a way to train a model on a video which is annotated instead of images? Is it an issue for darkflow or more for tensorflow?

Regards, Kris

Alemolet commented 5 years ago

If the video is annotated, you can split the video into frames and link to them their own annotations. I wrote a script to split the video into frames, save the annotations I needed in an array list and since the original video annotation was in .txt (YOLO output format), finally transformed them from .txt to .xml format (that DarkFlow needs in order to train). It was pretty sketchy but not a real big deal to be honest.

Alemolet commented 5 years ago

If you know what every number in the original annotation file mean, this job goes without saying :) Hope I gave you a good suggestion to start with.