search

ZhaoJ9014 / face.evoLVe

🔥🔥High-Performance Face Recognition Library on PaddlePaddle & PyTorch🔥🔥
MIT License
3.41k stars 759 forks source link

rgb conversion in perform_val #168

Open csookim opened 3 years ago

csookim commented 3 years ago

Isn't it right to add [:, [2, 1, 0], :, :] at the 199th line in utils.py?

https://github.com/ZhaoJ9014/face.evoLVe.PyTorch/blob/a33a9121198ed354eb6b0d7c214443f09908ccc1/util/utils.py#L188 https://github.com/ZhaoJ9014/face.evoLVe.PyTorch/blob/a33a9121198ed354eb6b0d7c214443f09908ccc1/util/utils.py#L199

I think rgb conversion is also needed to the tensors not included in the batch