if __name__ == "__main__":
test_data_visualizer = False
test_pre_processor = False
test_dicom_file_manager = False
test_file_manager = False
test_classifier_comparator = True
# ----------------------------------- FileManager -----------------------------------
if test_file_manager:
fm = FileManager()
training_path = 'covid19_repo/data/x_rays/covid'
shape = (64, 64)
shuffle = True
x, y = fm.get_classifier_image_training_data(
training_data_path=training_path, shape=shape, shuffle=shuffle)
print(x.shape, y.shape)
# ----------------------------------- DataVisualizer -----------------------------------
if test_data_visualizer:
visualize_pca = False
visualize_cnn_activations = True
# ---------- Visualizing PCA ---------
if visualize_pca:
n_classes = 3
x = np.random.random((100, 32, 32))
y = np.random.randint(0, n_classes, size=100)
dv = DataVisualizer(x=x, y=y)
