didadmin/docuseal
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

optimize detect fields

Browse Source
This commit is contained in:
Pete Matsyburka
2025-11-16 19:05:54 +02:00
parent 5fec952b8c
commit 1d87e210be
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/templates/image_to_fields.rb
+6 -6
View File
@@ -37,10 +37,10 @@ module Templates
transform_info[:trim_offset_x] = base_offset_x transform_info[:trim_offset_x] = base_offset_x
transform_info[:trim_offset_y] = base_offset_y + r[:offset_y] 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] boxes = outputs['dets'][0, true, true]
logits = Numo::SFloat.cast(outputs['labels'])[0, true, true] logits = outputs['labels'][0, true, true]
postprocess_outputs(boxes, logits, transform_info, acc, confidence:, temperature:, resolution:) postprocess_outputs(boxes, logits, transform_info, acc, confidence:, temperature:, resolution:)
end end
@@ -50,10 +50,10 @@ module Templates
transform_info[:trim_offset_x] = base_offset_x transform_info[:trim_offset_x] = base_offset_x
transform_info[:trim_offset_y] = base_offset_y 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] boxes = outputs['dets'][0, true, true]
logits = Numo::SFloat.cast(outputs['labels'])[0, true, true] logits = outputs['labels'][0, true, true]
detections = postprocess_outputs(boxes, logits, transform_info, confidence:, temperature:, resolution:) detections = postprocess_outputs(boxes, logits, transform_info, confidence:, temperature:, resolution:)
end end