PaddleClas icon indicating copy to clipboard operation
PaddleClas copied to clipboard

训练灰度图

Open Tolk5946 opened this issue 1 year ago • 6 comments

如果需要训练灰度图,请问怎么改配置文件呢

Tolk5946 avatar May 05 '23 09:05 Tolk5946

是训练阶段就是用单通道的输入吗?

zengshao0622 avatar May 05 '23 09:05 zengshao0622

是的 就是我的数据集就是单通道灰度图

Tolk5946 avatar May 05 '23 09:05 Tolk5946

如果训练阶段会将单通道的灰度图转成三通道的输入,可以将配置文件里的DataLoader.transform_ops.DecodeImage.backend设置为"pil"。如果训练阶段就使用单通道的输入,就需要自行实现对应的Dataloader了,并且组网代码可能也要修改。

zengshao0622 avatar May 05 '23 09:05 zengshao0622

噢噢,那如果将接口的backend设置为pil,对图片效果影响大吗?

Tolk5946 avatar May 05 '23 10:05 Tolk5946

而且好像接口DataLoader.transform_ops.DecodeImage下没有参数backend

Tolk5946 avatar May 05 '23 10:05 Tolk5946

而且似乎接口DataLoader.transform_ops.DecodeImage下没有参数backend 请问您训练灰度是否成功?

liujia761 avatar Jan 02 '24 07:01 liujia761