显卡RTX:3090
model:bisenet
输入图像:(4000,3000,3)
预处理:3.0 ms ;
其中:
resize (4000,3000)-->(512,512) :1.1 ms;
归一化:norm /255 :0.7 ms;
减均值 :-mean:1.0 ms;
除以方差:std:0.9
注意:减均值与除以方差,要写成:
image[:,:,0] -=self.mean[0]; image[:,:,1] -=self.mean[1]; image[:,:,2] -=self.mean[2]
不能写成:
image[:,:,0] /= self.std[0]; image[:,:,1] /= self.std[1]; image[:,:,2] /= self.std[2]
否则时间就约为3ms。
推理:7.0 ms;
后处理:3ms。包括如下操作
pred = output.data.cpu().numpy()
pred = np.argmax(pred, axis=1)[0]#得到每行
resize回原图:(512,512)--->(4000,300),1ms
opencv fincontours时间:
resize大小 resize/ms 时间 CV2.findcountours/ms时间
原图 0.0 5.2
256 0.4 5.0
512 0.5 4.9
1024 0.5 4.7
2048 0.6 6.4