arcface-pytorch
arcface-pytorch copied to clipboard
输入图片转灰度
dataset.py第48行: data = data.convert('L') 为什么输入图片要转成灰度的,保留RBG彩色三通道的不好嘛,难道就为了节省内存嘛
自然是可以的,只需要修改一下第一个卷积层的channel数。 我理解这样做唯一的好处是,模型要求输入单通道的灰度图的话,在后面测试的时候可以将这些单通道图concat起来类似mini-batch,进行批测试,使得每一个通道的输出都是每个样本的类别结果。如果模型输入RGB三通道图的话就没法concat,只能一张图一张图的测试。
如果只是做人脸识别的话,也要将数据预处理成灰度图吗
人脸识别没听说过必须转为灰度的操作。