当前位置:   article > 正文

pytorch ,ncnn,tnn等的减均值,归一化处理等_ncnn 归一化

ncnn 归一化

1 pytorch

一般通过 transform = transforms.Compose([ 
                transforms.ToTensor(),
                transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.50196078, 0.50196078, 0.50196078])])

toTensor()会进行归一化,Normalize的操作是(x-mean)/std

2 ncnn

const float mean_vals2[3] ={0.4,0.4,0.4};
const float std_vals2[3] = {4,4,4};
int.substract_mean_normalize(mean_vals2,std_vals2);

ncnn不会进行主动归一化,substract_mean_normalize的操作是(x-mean_vals2)*std_vals2

3 tnn

TNN_NS::MatConvertParam input_cvt_param; 
input_cvt_param.bias  = {1,1,1,0};
input_cvt_param.scale = {0,0,0,0};

auto status = instance_->SetInputMat(image_mat, input_cvt_param);

tnn不会进行主动归一化,input_cvt_param的操作是 x*input_cvt_param.scale+input_cvt_param.bias

车辆,车牌,反光衣,安全帽等数据集,链接,有兴趣的朋友可以看下 链接:https://pan.baidu.com/s/1ebCKOAE5Rx0PejiyRsTSXg 提取码:5555

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Monodyee/article/detail/703347
推荐阅读
相关标签
  

闽ICP备14008679号