未验证 提交 5bc3f230 编写于 作者: Y Yonghye Kwon 提交者: GitHub

chore(demo): add size adaptive blobFromImage function in TRT(#295)

chore(demo): add size adaptive blobFromImage function in TRT
上级 38e3c560
......@@ -211,8 +211,8 @@ float* blobFromImage(cv::Mat& img){
float* blob = new float[img.total()*3];
int channels = 3;
int img_h = 640;
int img_w = 640;
int img_h = img.rows;
int img_w = img.cols;
std::vector<float> mean = {0.485, 0.456, 0.406};
std::vector<float> std = {0.229, 0.224, 0.225};
for (size_t c = 0; c < channels; c++)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册