currently finished pipeline

main.py

@@ -0,0 +1,25 @@
+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)
+    

polygons_processing/postpprocess_detectree2.py

@@ -351,4 +351,6 @@ def postprocess(prediction_geojson_path):
 
     df_res = process([df])
 
-    export_df_as_geojson(df=df_res, filename="postprocessed_predictions")
+    export_df_as_geojson(df=df_res, filename="postprocessed_predictions")
+
+    return df_res