diff --git a/.rubocop.yml b/.rubocop.yml index aad5509e..294fc484 100644 --- a/.rubocop.yml +++ b/.rubocop.yml @@ -45,6 +45,9 @@ Metrics/PerceivedComplexity: Style/MultipleComparison: Enabled: false +Style/NumericPredicate: + Enabled: false + Naming/PredicateMethod: Enabled: false diff --git a/lib/templates/image_to_fields.rb b/lib/templates/image_to_fields.rb index fa342f4e..50837c1c 100755 --- a/lib/templates/image_to_fields.rb +++ b/lib/templates/image_to_fields.rb @@ -77,7 +77,7 @@ module Templates end def build_fields_from_detections(detections, image) - Array.new(detections[:xyxy].shape[0]) do |i| + detections[:xyxy].shape[0].times.filter_map do |i| x1 = detections[:xyxy][i, 0] y1 = detections[:xyxy][i, 1] x2 = detections[:xyxy][i, 2] @@ -92,6 +92,12 @@ module Templates x1_norm = x2 / image.width.to_f y1_norm = y2 / image.height.to_f + x1_norm = 1 if x1_norm > 1 + y1_norm = 1 if y1_norm > 1 + + next if x0_norm < 0 || x0_norm > 1 + next if y0_norm < 0 || y0_norm > 1 + type_name = ID_TO_CLASS[class_id] Field.new(