Micaela Verucchi
32 lines
732 B
C++
32 lines
732 B
C++
#ifndef BOUNDINGBOX_H
|
|
#define BOUNDINGBOX_H
|
|
|
|
#include <iostream>
|
|
#include "tkdnn.h"
|
|
|
|
namespace tk { namespace dnn {
|
|
class BoundingBox : public tk::dnn::box
|
|
{
|
|
float overlap(const float p1, const float l1, const float p2, const float l22);
|
|
float boxesIntersection(const BoundingBox &b);
|
|
float boxesUnion(const BoundingBox &b);
|
|
|
|
public:
|
|
|
|
int uniqueTruthIndex = -1;
|
|
int truthFlag = 0;
|
|
float maxIoU = 0;
|
|
|
|
float IoU(const BoundingBox &b);
|
|
void clear();
|
|
|
|
friend std::ostream& operator<<(std::ostream& os, const BoundingBox& bb);
|
|
};
|
|
|
|
std::ostream& operator<<(std::ostream& os, const BoundingBox& bb);
|
|
bool boxComparison (const BoundingBox& a,const BoundingBox& b) ;
|
|
|
|
}}
|
|
#endif /*BOUNDINGBOX_H*/
|
|
|