Infrared image enhancement with learned features