In [ ]:
%matplotlib inline

Simulating a dead CCD column

Example usage of the DeadColumn augmentation.


In [ ]:
import sys
import numpy as np
import matplotlib.pyplot as plt

from astronet.augmentations import DeadColumn

X = np.random.rand(4,1,50,50)*50
y = np.ones(4)

augment = DeadColumn(prob=1)
Xtransformed, _, _ = augment.apply(X, y, None)

fig, ax = plt.subplots(len(X),2, figsize=(3,6), subplot_kw={'xticks': [], 'yticks': []})
fig.subplots_adjust(wspace=0.1)
ax[0][0].set_title("Before")
ax[0][1].set_title("After")
for i in range(len(X)):
    ax[i][0].imshow(X[i][0])
    ax[i][1].imshow(Xtransformed[i][0])