main.py

from detectree2model.predictions.predict import run_detectree2
from polygons_processing.postpprocess_detectree2 import postprocess, export_df_as_geojson
from generate_tree_images.generate_tree_images import generate_tree_images
from classification.classification_predict import classify
import os

if __name__=="__main__":
    tif_input = ""
    detectree2_output = run_detectree2(tif_input)

    processed_output_df = postprocess(detectree2_output)
    processed_geojson = '../polygons_processing/postprocessed_predictions'

    generate_tree_images(processed_geojson, tif_input)
    output_folder = '../generate_tree_images/tree_images'

    for file_name in os.listdir(output_folder):
        file_path = os.path.join(output_folder, file_name)
        probs = classify(file_path)
        #add probs to df and then convert to geojson again for final output
        processed_output_df['class_probs'] = probs
    
    final_output_path = 'result.geojson'
    export_df_as_geojson(processed_output_df, final_output_path)