arcface-pytorch icon indicating copy to clipboard operation
arcface-pytorch copied to clipboard

输入图片转灰度

Open hpc203 opened this issue 4 years ago • 3 comments

dataset.py第48行: data = data.convert('L') 为什么输入图片要转成灰度的,保留RBG彩色三通道的不好嘛,难道就为了节省内存嘛

hpc203 avatar Sep 09 '20 10:09 hpc203

自然是可以的,只需要修改一下第一个卷积层的channel数。 我理解这样做唯一的好处是,模型要求输入单通道的灰度图的话,在后面测试的时候可以将这些单通道图concat起来类似mini-batch,进行批测试,使得每一个通道的输出都是每个样本的类别结果。如果模型输入RGB三通道图的话就没法concat,只能一张图一张图的测试。

Boatsure avatar Oct 13 '20 01:10 Boatsure

如果只是做人脸识别的话,也要将数据预处理成灰度图吗

wy321-wh48 avatar Jan 10 '22 07:01 wy321-wh48

人脸识别没听说过必须转为灰度的操作。

jdhao avatar Mar 08 '22 08:03 jdhao