Skip to content

Commit

Permalink
adjusting icy_resin threshold to 0.1
Browse files Browse the repository at this point in the history
  • Loading branch information
asaxton committed Nov 20, 2024
1 parent 9646e53 commit fb30b04
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/models/icy_resin_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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')
Expand Down

0 comments on commit fb30b04

Please sign in to comment.