search

isseu / emotion-recognition-neural-networks

Emotion recognition using DNN with tensorflow
MIT License
835 stars 308 forks source link

Bug in the dataset_loader.py file #14

Open dearhoper opened 7 years ago

dearhoper commented 7 years ago

Hi @isseu , There are two mistakes in the dataset_loader.py file. I wonder if you need to revise them. See Line 17 and Line 19. I think they should be changed to: self._images_test = self._images_test .reshape([-1, SIZE_FACE, SIZE_FACE, 1]) self._labels_test = self._labels_test .reshape([-1, len(EMOTIONS)])

jorgeribeiro commented 6 years ago

Yeah, I found it too. After this step my training set and validation set had the same length. In the cvs conversion everything was alright.