bug in file save_output

[dadamsncsa]

epoch: 0 (35/36), ['loss', 'acc']: [0.6927258, 0.70576286]
Traceback (most recent call last):
  File "/home/dadams2/.conda/envs/unet/lib/python3.5/site-packages/PIL/BmpImagePlugin.py", line 231, in _save
    rawmode, bits, colors = SAVE[im.mode]
KeyError: 'F'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "main.py", line 33, in <module>
    run.train()
  File "/local/6081513/unet-sem/master_adb1e74_07-24-2019-14:00:15/unet-sem-6081513/model.py", line 133, in train
    save_output(out_imgs[0,...],out_masks[0,...],prediction[0,...],index=i,epoch=epoch)
  File "/local/6081513/unet-sem/master_adb1e74_07-24-2019-14:00:15/unet-sem-6081513/utils.py", line 12, in save_output
    print_img.save(os.path.join(directory,'%d_epoch'%epoch,'%d_input_image.bmp'%index))
  File "/home/dadams2/.conda/envs/unet/lib/python3.5/site-packages/PIL/Image.py", line 1950, in save
    save_handler(self, fp, filename)
  File "/home/dadams2/.conda/envs/unet/lib/python3.5/site-packages/PIL/BmpImagePlugin.py", line 233, in _save
    raise IOError("cannot write mode %s as BMP" % im.mode)
OSError: cannot write mode F as BMP

[dadamsncsa]

Traceback (most recent call last):
  File "/home/dadams2/.conda/envs/unet/lib/python3.5/site-packages/PIL/PngImagePlugin.py", line 726, in _save
    rawmode, mode = _OUTMODES[mode]
KeyError: 'F'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "main.py", line 33, in <module>
    run.train()
  File "/local/6084871/unet-sem/master_21d245a_07-25-2019-10:45:04/unet-sem-6084871/model.py", line 133, in train
    save_output(out_imgs[0,...],out_masks[0,...],prediction[0,...],index=i,epoch=epoch)
  File "/local/6084871/unet-sem/master_21d245a_07-25-2019-10:45:04/unet-sem-6084871/utils.py", line 12, in save_output
    print_img.save(os.path.join(directory,'%d_epoch'%epoch,'%d_input_image.png'%index))
  File "/home/dadams2/.conda/envs/unet/lib/python3.5/site-packages/PIL/Image.py", line 1950, in save
    save_handler(self, fp, filename)
  File "/home/dadams2/.conda/envs/unet/lib/python3.5/site-packages/PIL/PngImagePlugin.py", line 728, in _save
    raise IOError("cannot write mode %s as PNG" % mode)
OSError: cannot write mode F as PNG