This code is currently functional but not giving good results. I had to convert to grayscale due to weird matmul errors.
