Crater Detection Pipeline

Pipeline and Image Objects

class pycda.CDA(detector='unet', extractor='circle', classifier='convolution')

Bases: object

the CDA class is a pipeline that makes predictions on an input image by passing it through a series of models. For each prediction, it generates a prediction object that tracks the outputs of the various models throughout the process.

get_prediction(input_image)

Used for accessing the prediction object. Calls predictions and returns the prediction object for advanced statistics and visualizations.

predict(input_image, threshold=0.5)

Intended for ‘out of box’ use. Calls predictions and returns a pandas dataframe with crater predictions.

pycda.load_image(filename)

load an image from input filepath and return a numpy array image.