Darknet

Darknet Detector Algorithm

class darknet_detector : public kwiver::vital::algo::image_object_detector

Public Functions

virtual bool check_configuration(vital::config_block_sptr config) const override

Check that the algorithm’s configuration config_block is valid

This checks solely within the provided config_block and not against the current state of the instance. This isn’t static for inheritance reasons.

Parameters:

config – The config block to check configuration of.

Returns:

true if the configuration check passed and false if it didn’t.

virtual vital::detected_object_set_sptr detect(vital::image_container_sptr image_data) const override

Find all objects on the provided image

This method analyzes the supplied image and along with any saved context, returns a vector of detected image objects.

Parameters:

image_data – the image pixels

Returns:

vector of image objects found

class priv

Darknet Trainer Algorithm

class darknet_trainer : public kwiver::vital::algo::train_detector

Darknet Training Utility Class.

Public Functions

virtual bool check_configuration(vital::config_block_sptr config) const override

Check that the algorithm’s configuration config_block is valid

This checks solely within the provided config_block and not against the current state of the instance. This isn’t static for inheritance reasons.

Parameters:

config – The config block to check configuration of.

Returns:

true if the configuration check passed and false if it didn’t.

virtual void train_from_disk(vital::category_hierarchy_sptr object_labels, std::vector<std::string> train_image_names, std::vector<vital::detected_object_set_sptr> train_groundtruth, std::vector<std::string> test_image_names, std::vector<vital::detected_object_set_sptr> test_groundtruth) override

Train a detection model given a list of images and detections

This varient is geared towards offline training.

Parameters:

• object_labels – object category labels for training

• train_image_list – list of train image filenames

• train_groundtruth – annotations loaded for each image

• test_image_list – list of test image filenames

• test_groundtruth – annotations loaded for each image

class priv

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)