From fb30b0405374c02f08d458c15f0d7e578632bc78 Mon Sep 17 00:00:00 2001 From: Aaron Saxton Date: Wed, 20 Nov 2024 12:26:34 -0600 Subject: [PATCH] adjusting icy_resin threshold to 0.1 --- src/models/icy_resin_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/models/icy_resin_model.py b/src/models/icy_resin_model.py index 3a56d4a..e8b86eb 100644 --- a/src/models/icy_resin_model.py +++ b/src/models/icy_resin_model.py @@ -125,7 +125,7 @@ def inference(self, image, legend_images, data_id=-1): # pipeline_manager.log(logging.DEBUG, "\t\tExecution time for {} legend: {:.2f} seconds. {:.2f} patches per second".format(label, lgd_time, (rows*cols)/lgd_time)) # Minimum confidence threshold for a prediction - map_prediction[map_confidence < 0.333] = 0 + map_prediction[map_confidence < 0.1] = 0 # For profiling memory usage # torch.cuda.memory._dump_snapshot(f'gpu_snapshots/{data_id}_inference.pickle')