Description
Great job with fixthejet. From your blog website I noticed that your tool does not work well with images which have black and white (try example below).
Although you make a note of that above the output, I think it will be better if you can make it more robust. Specifically because almost all images have some amount of black or white. I strongly suggest using a regression tree (like scipy.spatial.KDTree) to ignore the pixels not in jet colormap. You can check out a variant of that implementation here https://github.com/kkraoj/fixthejet.git . If you still have time to work on this, can you please improve it? Fixing the jet is a big necessaity in my community (Earth Sciences) so it would be pretty cool if you could update your website with an improved version. I can also help in and want no recognition.
Edit: To be clear, the script referenced in my profile does work on images such as the one posted aabove.