[attack] Fix box variation for empty bounding boxes

wuhanstudio · Feb 6, 2024 · 39d5741 · 39d5741
1 parent 605f7e3
commit 39d5741
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/examples/attack/yolov3_pcb_attack_evaluation.py b/examples/attack/yolov3_pcb_attack_evaluation.py
@@ -106,9 +106,11 @@
                     indexes = indexes.flatten()
                     tb.log_scalar('box variation', (len(boxes) + len(last_boxes) - len(indexes)) / len(boxes), n)
             elif last_boxes.size == 0 and boxes.size == 0:
-                tb.log_scalar('box variation', 0.0, n)
-            else:
+                # No bounding boxes, all consistent
                 tb.log_scalar('box variation', 1.0, n)
+            else:
+                # Either one is empty, none consistent
+                tb.log_scalar('box variation', 0.0, n)
         else:
             tb.log_scalar('box variation', 1.0, n)
 

diff --git a/examples/attack/yolov3_tog_attack_evaluation.py b/examples/attack/yolov3_tog_attack_evaluation.py
@@ -99,9 +99,11 @@
                     indexes = indexes.flatten()
                     tb.log_scalar('box variation', (len(boxes) + len(last_boxes) - len(indexes)) / len(boxes), n)
             elif last_boxes.size == 0 and boxes.size == 0:
-                tb.log_scalar('box variation', 0.0, n)
-            else:
+                # No bounding boxes, all consistent
                 tb.log_scalar('box variation', 1.0, n)
+            else:
+                # Either one is empty, none consistent
+                tb.log_scalar('box variation', 0.0, n)
         else:
             tb.log_scalar('box variation', 1.0, n)