PaddleX icon indicating copy to clipboard operation
PaddleX copied to clipboard

下载的模型目前不支持单通道图像的训练和预测

Open qing130 opened this issue 4 years ago • 2 comments

目前PaddleX下载的模型似乎只支持三通道的图像,而工业场景有些用灰度图即可解决,如果用RGB图像则运行较长。 请问可以在模型API加个输入通道数的参数吗?(当然,模型应该也要做相应的更改)。

qing130 avatar Nov 17 '21 07:11 qing130

目前PaddleX 2.0默认会把灰度图转成三通道进行处理。

如果用RGB图像则运行较长。

这个我们之前也测试过,模型层面只会影响第一个conv,这个对性能影响可以忽略不计。如果图片不是超大的那种,预处理环节的也没有太大影响。PaddleX 1.3.11支持任意通道的数据,您可以先用1.3.11的试试看。

FlyingQianMM avatar Nov 18 '21 02:11 FlyingQianMM

按理单通道的话,应该所有的网络层的参数都要减掉一些了。单纯改输入通道,确实对运算量影响不大

qing130 avatar Nov 30 '21 09:11 qing130