I downloaded your pretrained models and got your code working on my macbook with great results. However, when I did some experimentation I noticed the colors of the image_stimuli in the predict function where wrong. I then found that when the images are being loaded on line 17 in 03-predict.py you seem to shuffle the color channels:
cv2.cvtColor( ... , cv2.COLOR_BGR2RGB)
What is the reason for doing this? Have your pretrained models been trained using shuffled image channels too?
I downloaded your pretrained models and got your code working on my macbook with great results. However, when I did some experimentation I noticed the colors of the
image_stimuli
in thepredict
function where wrong. I then found that when the images are being loaded on line 17 in03-predict.py
you seem to shuffle the color channels:cv2.cvtColor( ... , cv2.COLOR_BGR2RGB)
What is the reason for doing this? Have your pretrained models been trained using shuffled image channels too?