На самом деле для одноканального изображения дальнейшая проверка условия излишняя. Поэтому прямо в условии можно сделать оставшуюся операцию для выходного изображения и выйти из программы.