util.h

Go to the documentation of this file.

 
#ifndef _NNDEPLOY_DETECT_DETECT_UTIL_H_
#define _NNDEPLOY_DETECT_DETECT_UTIL_H_
 
#include "nndeploy/base/any.h"
#include "nndeploy/base/common.h"
#include "nndeploy/base/glic_stl_include.h"
#include "nndeploy/base/log.h"
#include "nndeploy/base/macro.h"
#include "nndeploy/base/object.h"
#include "nndeploy/base/param.h"
#include "nndeploy/base/status.h"
#include "nndeploy/base/string.h"
#include "nndeploy/base/type.h"
#include "nndeploy/detect/result.h"
#include "nndeploy/device/buffer.h"
#include "nndeploy/device/device.h"
#include "nndeploy/device/memory_pool.h"
#include "nndeploy/device/tensor.h"
 
namespace nndeploy {
namespace detect {
 
extern NNDEPLOY_CC_API std::array<float, 4> getOriginBox(
    float xmin, float ymin, float xmax, float ymax, const float *scale_factor,
    float x_offset, float y_offset, int ori_width, int ori_height);
 
extern NNDEPLOY_CC_API std::array<float, 4> getOriginBox(
    const std::array<float, 4> &box, const float *scale_factor, float x_offset,
    float y_offset, int ori_width, int ori_height);
 
float computeIOU(float xmin0, float ymin0, float xmax0, float ymax0,
                 float xmin1, float ymin1, float xmax1, float ymax1);
 
float computeIOU(const std::array<float, 4> &box0,
                 const std::array<float, 4> &box1);
 
float computeIOU(const float *boxes, int i, int j);
 
base::Status computeNMS(const DetectResult &src, std::vector<int> &keep_idxs,
                        const float iou_threshold);
 
base::Status fastNMS(const DetectResult &src, std::vector<int> &keep_idxs,
                     const float iou_threshold);
 
}  // namespace detect
}  // namespace nndeploy
 
#endif /* _NNDEPLOY_DETECT_DETECT_COMMON_H_ */