From 497a6f2401deaeb945495bc9b2755c25b6ba64ed Mon Sep 17 00:00:00 2001 From: Jona Date: Mon, 30 Oct 2023 20:29:46 -0400 Subject: [PATCH] Add debug prints, and try reversing UNET/MASKRCNN? --- cerulean_cloud/cloud_run_orchestrator/handler.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/cerulean_cloud/cloud_run_orchestrator/handler.py b/cerulean_cloud/cloud_run_orchestrator/handler.py index 63e0e758..4bc899bd 100644 --- a/cerulean_cloud/cloud_run_orchestrator/handler.py +++ b/cerulean_cloud/cloud_run_orchestrator/handler.py @@ -407,8 +407,20 @@ async def _orchestrate( 20, "offset tiles", ) + print("base_tiles_inference", base_tiles_inference) + print("base_tiles_inference[0]", base_tiles_inference[0]) + print("base_tiles_inference[0].stack", base_tiles_inference[0].stack) + print("base_tiles_inference[0].stack[0]", base_tiles_inference[0].stack[0]) + print( + "base_tiles_inference[0].stack[0].dict()", + base_tiles_inference[0].stack[0].dict(), + ) + print( + "base_tiles_inference[0].stack[0].dict().get(classes)", + base_tiles_inference[0].stack[0].dict().get("classes"), + ) - if model.type == "MASKRCNN": + if model.type == "UNET": print("Loading all tiles into memory for merge!") ds_base_tiles = [] for base_tile_inference in base_tiles_inference: @@ -459,7 +471,7 @@ async def _orchestrate( dst.write(ar) out_fc_offset = get_fc_from_raster(offset_tile_inference_file) - elif model.type == "UNET": + elif model.type == "MASKRCNN": # out_fc = geojson.FeatureCollection( # features=flatten_feature_list(base_tiles_inference) # )