| 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) | |