yolov3 Namespace Reference

Classes
struct	Box
	Box representation structure. More...
struct	Detection
	Detection structure. More...
struct	NMSConfig
	Non Maxima Suprresion configuration meta-data. More...

Functions
void	print_detection (std::ostream &os, const std::vector< Detection > &detections)
	Print identified yolo detections. More...
std::vector< Detection >	nms (const NMSConfig &config, const std::vector< float > &detected_boxes)
	Perform Non-Maxima Supression on a list of given detections. More...

Function Documentation

nms()

std::vector< Detection > nms	(	const NMSConfig &	config,
		const std::vector< float > &	detected_boxes
	)

Perform Non-Maxima Supression on a list of given detections.

Parameters

[in]	config	Configuration metadata for NMS
[in]	detected_boxes	Detected boxes

Returns

A vector with the final detections

Definition at line 100 of file NMS.cpp.

References Detection::classes, NMSConfig::iou_threshold, and NMSConfig::num_classes.

                                                                   {
    // Get detections that comply with the expected confidence threshold
    std::vector<Detection> detections =
        convert_to_detections(config, detected_boxes);

    const unsigned int num_detections = static_cast<unsigned int>(detections.size());
    for (unsigned int c = 0; c < config.num_classes; ++c)
    {
        // Sort classes
        std::sort(detections.begin(), detections.begin() + static_cast<std::ptrdiff_t>(num_detections),
                  [c](Detection& detection1, Detection& detection2)
                    {
                        return (detection1.classes[c] - detection2.classes[c]) > 0;
                    });
        // Clear detections with high IoU
        for (unsigned int d = 0; d < num_detections; ++d)
        {
            // Check if class is already cleared/invalidated
            if (detections[d].classes[c] == 0.f)
            {
                continue;
            }

            // Filter out boxes on IoU threshold
            const Box& box1 = detections[d].box;
            for (unsigned int b = d + 1; b < num_detections; ++b)
            {
                const Box& box2 = detections[b].box;
                if (iou(box1, box2) > config.iou_threshold)
                {
                    detections[b].classes[c] = 0.f;
                }
            }
        }
    }
    return detections;
}

print_detection()

void print_detection	(	std::ostream &	os,
		const std::vector< Detection > &	detections
	)

Print identified yolo detections.

Parameters

[in,out]	os	Output stream to print to
[in]	detections	Detections to print

Definition at line 83 of file NMS.cpp.

{
    for (const auto& detection : detections)
    {
        for (unsigned int c = 0; c < detection.classes.size(); ++c)
        {
            if (detection.classes[c] != 0.0f)
            {
                os << c << " " << detection.classes[c] << " " << detection.box.xmin
                   << " " << detection.box.ymin << " " << detection.box.xmax << " "
                   << detection.box.ymax << std::endl;
            }
        }
    }
}