Darknet

Darknet is an open source neural network framework written in C and CUDA.

The following algorithm implementations use Darknet

darknet_detector
class kwiver::arrows::darknet::darknet_detector

Inherits from kwiver::vital::algorithm_impl< darknet_detector, vital::algo::image_object_detector >

darknet_trainer
class kwiver::arrows::darknet::darknet_trainer

Darknet Training Utility Class.

Inherits from kwiver::vital::algorithm_impl< darknet_trainer, vital::algo::train_detector >

In the pipe files, you can tune the algorithm with these variables :
  • darknet:thresh
  • darknet:hier_thresh
  • darknet:gpu_index

FAQ

I am running out of memory in CUDA...
Try one or both of these suggestions: - Change the darknet/models/virat.cfg variables height,weight to smaller powers of 32 - Change the darknet/models/virat.cfg variables batch and subdivisions (make sure they are still the same)