FastDeploy
latest
Fast & Easy to Deploy!
|
Preprocessor object for YOLOv5Cls serials model. More...
#include <preprocessor.h>
Public Member Functions | |
YOLOv5ClsPreprocessor () | |
Create a preprocessor instance for YOLOv5Cls serials model. | |
bool | Run (std::vector< FDMat > *images, std::vector< FDTensor > *outputs, std::vector< std::map< std::string, std::array< float, 2 >>> *ims_info) |
Process the input image and prepare input tensors for runtime. More... | |
void | SetSize (const std::vector< int > &size) |
Set target size, tuple of (width, height), default size = {224, 224}. | |
std::vector< int > | GetSize () const |
Get target size, tuple of (width, height), default size = {224, 224}. | |
Preprocessor object for YOLOv5Cls serials model.
bool fastdeploy::vision::classification::YOLOv5ClsPreprocessor::Run | ( | std::vector< FDMat > * | images, |
std::vector< FDTensor > * | outputs, | ||
std::vector< std::map< std::string, std::array< float, 2 >>> * | ims_info | ||
) |
Process the input image and prepare input tensors for runtime.
[in] | images | The input image data list, all the elements are returned by cv::imread() |
[in] | outputs | The output tensors which will feed in runtime |
[in] | ims_info | The shape info list, record input_shape and output_shape |