Попытка перекрыть изображения TESS и 2MASS с помощью WCSaxes и Astroquery

Question

Я пытаюсь построить изображение 2MASS на той же проекции, что и изображение TESS. Раньше я делал это с pywcsgrid2, но я больше не могу его установить. Так что я пытаюсь с Astropy WCSAxes.

И изображение TESS, и изображение 2MASS извлекаются с помощью функций Astroquery (TESScut и SkyView соответственно). Я могу создать график каждого изображения индивидуально на соответствующих осях WCS. Но когда я пытаюсь нанести изображение 2MASS на ось с помощью TESS WCS (либо как контур, либо как само изображение), оно сжимает изображение до нижнего левого угла. Может ли кто-нибудь сказать мне, что я делаю не так, или что-то не так с WCS.

      import numpy as np
import matplotlib.pyplot as plt
from matplotlib import colors
from astroquery.skyview import SkyView
from astropy.wcs import WCS
import astropy.io.fits as fits
import astropy.units as u
from astropy.coordinates import SkyCoord
from astroquery.mast import Tesscut

test_tic = 113449635
cutout_coord = SkyCoord(113.29538299104,-35.48102008076,unit=u.degree)
dir_ffi = "./temp/"
tess_data = Tesscut.download_cutouts(cutout_coord, size=20, path=dir_ffi)

tess_file = tess_data[2][0]

def extract_wcs(dataheader):
    """
    Generate WCS for a TESS or K2 TPF. dataheader is the header of extension #1
    """
    w5 = WCS(naxis=2)

    w5.wcs.crpix = [dataheader["1CRPX5"],dataheader["2CRPX5"]]
    w5.wcs.cdelt = np.array([dataheader["1CDLT5"],dataheader["2CDLT5"]])
    w5.wcs.crval = [dataheader["1CRVL5"],dataheader["2CRVL5"]]
    w5.wcs.ctype = [dataheader["1CTYP5"],dataheader["2CTYP5"]]
    w5.wcs.pc = [[dataheader["11PC5"],dataheader["12PC5"]],
                 [dataheader["21PC5"],dataheader["22PC5"]]]
    return w5


with fits.open(tess_file) as hdu:

    # not a coadd, but reducing code for the sake of the MWE
    coadd = hdu[1].data["FLUX"][100]
    dataheader = hdu[1].header
    w3 = extract_wcs(dataheader)

# This works just fine
plt.subplot(projection=w3)
plt.imshow(coadd,origin="lower", norm=colors.LogNorm())
plt.xlabel('RA')
plt.ylabel('Dec')
plt.show()

# Get 2MASS K image
twomass_images = SkyView.get_images(position=cutout_coord, survey=['2MASS-K'],pixels=500)
pix_2mass = twomass_images[0][0].data
hdr_2mass = twomass_images[0][0].header
wcs_2mass = WCS(hdr_2mass)

print(w3)
print(wcs_2mass)

# Try to plot the TESS image in the background, with 2MASS overlaid as a contour
# This yields oversized axes with the image in the lower left corner
ax = plt.subplot(projection=w3)
ax.imshow(coadd,origin="lower", norm=colors.LogNorm(),zorder=-10)
ax.contour(pix_2mass, transform=ax.get_transform(wcs_2mass),
           colors='k',zorder=5)
plt.xlabel('RA')
plt.ylabel('Dec')
plt.show()


# Try to plot the 2MASS image in the background, on the TESS WCS
# This yields oversized axes with the image in the lower left corner
ax = plt.subplot(projection=w3)
ax.imshow(pix_2mass, origin="lower", norm=colors.LogNorm(), zorder=-10,
          transform=ax.get_transform(wcs_2mass))
plt.xlabel('RA')
plt.ylabel('Dec')
plt.show()