|
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 |
1.8.13