polarisZhao / PFLD-pytorch

PFLD pytorch Implementation
798 stars 197 forks source link

模型训练时输入是BGR,使用camera时输入是RGB? #20

Closed JinBuLanG closed 4 years ago

JinBuLanG commented 4 years ago

训练时读入图片使用opencv,所以训练使用bgr,这是否会对结果造成影响?(我觉得不会)

gjd2017 commented 4 years ago

模型是BGR ,camera.py 中的将“input = cv2.cvtColor(input, cv2.COLOR_BGR2RGB)”注释掉,另外,这个版本的效果确实不太好,不过还是很感谢作者的开源代码。