Convert FIT to TIFF Python

fit_to_tiff.py

from astropy.io import fits

from PIL import Image

import numpy

import sys
import os

filepath = 'test.fit'
name, ext = os.path.splitext(filepath)

img = fits.open(filepath)

img.info()

data = img[0].data

width = data.shape[1]
height = data.shape[0]

outputArray = numpy.array(data, dtype=numpy.int16)

output = Image.fromarray(outputArray.reshape((height, width)), "I;16")

output.save(name + ".tif")