FastDeploy
latest
Fast & Easy to Deploy!
|
Postprocessor object for YOLOv7 serials model. More...
#include <postprocessor.h>
Public Member Functions | |
YOLOv7Postprocessor () | |
Create a postprocessor instance for YOLOv7 serials model. | |
bool | Run (const std::vector< FDTensor > &tensors, std::vector< DetectionResult > *results, const std::vector< std::map< std::string, std::array< float, 2 >>> &ims_info) |
Process the result of runtime and fill to DetectionResult structure. More... | |
void | SetConfThreshold (const float &conf_threshold) |
Set conf_threshold, default 0.25. | |
float | GetConfThreshold () const |
Get conf_threshold, default 0.25. | |
void | SetNMSThreshold (const float &nms_threshold) |
Set nms_threshold, default 0.5. | |
float | GetNMSThreshold () const |
Get nms_threshold, default 0.5. | |
Postprocessor object for YOLOv7 serials model.
bool fastdeploy::vision::detection::YOLOv7Postprocessor::Run | ( | const std::vector< FDTensor > & | tensors, |
std::vector< DetectionResult > * | results, | ||
const std::vector< std::map< std::string, std::array< float, 2 >>> & | ims_info | ||
) |
Process the result of runtime and fill to DetectionResult structure.
[in] | tensors | The inference result from runtime |
[in] | result | The output result of detection |
[in] | ims_info | The shape info list, record input_shape and output_shape |