在进行推理过程中,对送入到模型中的图片主要有两种处理措施:
一种是对原图大于模型输入的图片,进行压缩,这个是不可避免的
这里,我们只讨论小图片处理的情况
小图片处理措施
另外一种是对小于模型的输入图片的处理
- 创建一个跟模型输入要求一致的画布,将原图内容复制到画布的左上角,其他地方填充黑色或者灰色
- 对原图进行拉伸操作
两种方式,在处理时长上并没有明显差异。且第一种填充的方式,并不会改变原图的像素信息,为第一选择。
推理测试数据
测试平台
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.33.01 Driver Version: 440.33.01 CUDA Version: 10.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 TITAN RTX Off | 00000000:19:00.0 Off | N/A |
| 23% 51C P0 56W / 280W | 0MiB / 24220MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| No running processes found |