ronghuaiyang / arcface-pytorch

1.74k stars 392 forks source link

输入图片转灰度 #74

Open hpc203 opened 4 years ago

hpc203 commented 4 years ago

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

Boatsure commented 3 years ago

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

wy321-wh48 commented 2 years ago

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

jdhao commented 2 years ago

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