|
FastDeploy
latest
Fast & Easy to Deploy!
|
Postprocessor object for YOLOv5Cls serials model. More...
#include <postprocessor.h>
Public Member Functions | |
| YOLOv5ClsPostprocessor () | |
| Create a postprocessor instance for YOLOv5Cls serials model. | |
| bool | Run (const std::vector< FDTensor > &tensors, std::vector< ClassifyResult > *results, const std::vector< std::map< std::string, std::array< float, 2 >>> &ims_info) |
| Process the result of runtime and fill to ClassifyResult structure. More... | |
| void | SetTopK (const int &topk) |
| Set topk, default 1. | |
| float | GetTopK () const |
| Get topk, default 1. | |
Postprocessor object for YOLOv5Cls serials model.
| bool fastdeploy::vision::classification::YOLOv5ClsPostprocessor::Run | ( | const std::vector< FDTensor > & | tensors, |
| std::vector< ClassifyResult > * | results, | ||
| const std::vector< std::map< std::string, std::array< float, 2 >>> & | ims_info | ||
| ) |
Process the result of runtime and fill to ClassifyResult structure.
| [in] | tensors | The inference result from runtime |
| [in] | result | The output result of classification |
| [in] | ims_info | The shape info list, record input_shape and output_shape |
1.8.13