From 1d87e210be7f1471dc2cc76d2800e0e8de4e67c1 Mon Sep 17 00:00:00 2001 From: Pete Matsyburka Date: Sun, 16 Nov 2025 19:05:54 +0200 Subject: [PATCH] optimize detect fields --- lib/templates/image_to_fields.rb | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/lib/templates/image_to_fields.rb b/lib/templates/image_to_fields.rb index 50837c1c..786e9785 100755 --- a/lib/templates/image_to_fields.rb +++ b/lib/templates/image_to_fields.rb @@ -37,10 +37,10 @@ module Templates transform_info[:trim_offset_x] = base_offset_x transform_info[:trim_offset_y] = base_offset_y + r[:offset_y] - outputs = model.predict({ 'input' => input_tensor }) + outputs = model.predict({ 'input' => input_tensor }, output_type: :numo) - boxes = Numo::SFloat.cast(outputs['dets'])[0, true, true] - logits = Numo::SFloat.cast(outputs['labels'])[0, true, true] + boxes = outputs['dets'][0, true, true] + logits = outputs['labels'][0, true, true] postprocess_outputs(boxes, logits, transform_info, acc, confidence:, temperature:, resolution:) end @@ -50,10 +50,10 @@ module Templates transform_info[:trim_offset_x] = base_offset_x transform_info[:trim_offset_y] = base_offset_y - outputs = model.predict({ 'input' => input_tensor }) + outputs = model.predict({ 'input' => input_tensor }, output_type: :numo) - boxes = Numo::SFloat.cast(outputs['dets'])[0, true, true] - logits = Numo::SFloat.cast(outputs['labels'])[0, true, true] + boxes = outputs['dets'][0, true, true] + logits = outputs['labels'][0, true, true] detections = postprocess_outputs(boxes, logits, transform_info, confidence:, temperature:, resolution:) end