Darknet/YOLO v3.0-149-gb11c9d5
Object Detection Framework
 
Loading...
Searching...
No Matches
darknet_05_process_videos_multithreaded.cpp File Reference

This application will process one or more videos as fast as possible using multiple threads and save a new output video to disk. More...

Include dependency graph for darknet_05_process_videos_multithreaded.cpp:

Classes

struct  Frame
 Everything we know about a specific video frame is stored in one of these objects. More...
 

Functions

void detection_thread (size_t &total_objects_found)
 
int main (int argc, char *argv[])
 
void output_thread (cv::VideoWriter &out)
 
void resize_thread ()
 

Variables

bool all_threads_must_exit = false
 if something goes wrong, this flag gets set to true
 
std::set< Frameframes_waiting_for_output
 once a frame has been predicted, it is stored here
 
std::set< Frameframes_waiting_for_prediction
 once a frame has been resized, it is stored here
 
std::set< Frameframes_waiting_for_resize
 once a frame is read from the video, it is stored here
 
Darknet::NetworkPtr net = nullptr
 Darknet/YOLO neural network pointer.
 
cv::Size network_dimensions
 dimensions of the neural network that was loaded
 
size_t output_thread_starved = 0
 
std::chrono::high_resolution_clock::duration output_work_duration
 amount of time spent on the output video
 
size_t predict_thread_starved = 0
 
std::chrono::high_resolution_clock::duration predict_work_duration
 amount of time spent predicting frames
 
size_t reader_must_pause = 0
 
std::chrono::high_resolution_clock::duration reader_work_duration
 amount of time spent reading frames
 
size_t resize_thread_starved = 0
 
std::chrono::high_resolution_clock::duration resize_work_duration
 amount of time spent resizing frames
 
std::mutex waiting_for_output
 mutex to protect access to frames_waiting_for_output
 
std::mutex waiting_for_prediction
 mutex to protect access to frames_waiting_for_prediction
 
std::mutex waiting_for_resize
 mutex to protect access to frames_waiting_for_resize
 

Detailed Description

This application will process one or more videos as fast as possible using multiple threads and save a new output video to disk.

The results are not shown to the user. Call it like this:

darknet_05_process_videos_multithreaded LegoGears DSCN1582A.MOV

The output should be similar to this:

processing DSCN1582A.MOV:
-> total number of CPUs ..... 16
-> threads for this video ... 4
-> neural network size ...... 224 x 160 x 3
-> input video dimensions ... 640 x 480
-> input video frame count .. 1230
-> input video frame rate ... 29.970030 FPS
-> input video length ....... 41041 milliseconds
-> output filename .......... DSCN1582A_output.m4v
-> total frames processed ... 1230
-> time to process video .... 1719 milliseconds
-> processed frame rate ..... 715.532286 FPS
-> total objects founds ..... 6189
-> average objects/frame .... 5.031707

Function Documentation

◆ detection_thread()

void detection_thread ( size_t &  total_objects_found)
Here is the call graph for this function:
Here is the caller graph for this function:

◆ main()

int main ( int  argc,
char *  argv[] 
)
Here is the call graph for this function:

◆ output_thread()

void output_thread ( cv::VideoWriter &  out)
Here is the caller graph for this function:

◆ resize_thread()

void resize_thread ( )
Here is the call graph for this function:
Here is the caller graph for this function:

Variable Documentation

◆ all_threads_must_exit

bool all_threads_must_exit = false

if something goes wrong, this flag gets set to true

◆ frames_waiting_for_output

std::set<Frame> frames_waiting_for_output

once a frame has been predicted, it is stored here

◆ frames_waiting_for_prediction

std::set<Frame> frames_waiting_for_prediction

once a frame has been resized, it is stored here

◆ frames_waiting_for_resize

std::set<Frame> frames_waiting_for_resize

once a frame is read from the video, it is stored here

◆ net

Darknet::NetworkPtr net = nullptr

Darknet/YOLO neural network pointer.

◆ network_dimensions

cv::Size network_dimensions

dimensions of the neural network that was loaded

◆ output_thread_starved

size_t output_thread_starved = 0

◆ output_work_duration

std::chrono::high_resolution_clock::duration output_work_duration

amount of time spent on the output video

◆ predict_thread_starved

size_t predict_thread_starved = 0

◆ predict_work_duration

std::chrono::high_resolution_clock::duration predict_work_duration

amount of time spent predicting frames

◆ reader_must_pause

size_t reader_must_pause = 0

◆ reader_work_duration

std::chrono::high_resolution_clock::duration reader_work_duration

amount of time spent reading frames

◆ resize_thread_starved

size_t resize_thread_starved = 0

◆ resize_work_duration

std::chrono::high_resolution_clock::duration resize_work_duration

amount of time spent resizing frames

◆ waiting_for_output

std::mutex waiting_for_output

mutex to protect access to frames_waiting_for_output

◆ waiting_for_prediction

std::mutex waiting_for_prediction

mutex to protect access to frames_waiting_for_prediction

◆ waiting_for_resize

std::mutex waiting_for_resize

mutex to protect access to frames_waiting_for_resize