Format changed files with black==21.12b0

Signed-off-by: Foxglove144 <[email protected]>

Author: Foxglove144

art/attacks/evasion/adversarial_patch/adversarial_patch_pytorch.py

@@ -123,7 +123,9 @@ def __init__(
 
         torch_version = list(map(int, torch.__version__.lower().split("+", maxsplit=1)[0].split(".")))
         torchvision_version = list(map(int, torchvision.__version__.lower().split("+", maxsplit=1)[0].split(".")))
-        assert torch_version[0] >= 1 and torch_version[1] >= 7 or (torch_version[0]>=2), "AdversarialPatchPyTorch requires torch>=1.7.0"
+        assert (
+            torch_version[0] >= 1 and torch_version[1] >= 7 or (torch_version[0] >= 2)
+        ), "AdversarialPatchPyTorch requires torch>=1.7.0"
         assert (
             torchvision_version[0] >= 0 and torchvision_version[1] >= 8
         ), "AdversarialPatchPyTorch requires torchvision>=0.8.0"

examples/application_object_detection.py

@@ -130,11 +130,23 @@ def plot_image_with_boxes(img, boxes, pred_cls):
     for i in range(len(boxes)):
         # Draw Rectangle with the coordinates
 
-        cv2.rectangle(img, (int(boxes[i][0][0]), int(boxes[i][0][1])), (int(boxes[i][1][0]), int(boxes[i][1][1])),
-                      color=(0, 255, 0), thickness=rect_th)
+        cv2.rectangle(
+            img,
+            (int(boxes[i][0][0]), int(boxes[i][0][1])),
+            (int(boxes[i][1][0]), int(boxes[i][1][1])),
+            color=(0, 255, 0),
+            thickness=rect_th,
+        )
         # Write the prediction class
-        cv2.putText(img, pred_cls[i], (int(boxes[i][0][0]), int(boxes[i][0][1])), cv2.FONT_HERSHEY_SIMPLEX, text_size,
-                    (0, 255, 0), thickness=text_th)
+        cv2.putText(
+            img,
+            pred_cls[i],
+            (int(boxes[i][0][0]), int(boxes[i][0][1])),
+            cv2.FONT_HERSHEY_SIMPLEX,
+            text_size,
+            (0, 255, 0),
+            thickness=text_th,
+        )
     plt.axis("off")
     plt.imshow(img.astype(np.uint8), interpolation="nearest")
     plt.show()
@@ -206,4 +218,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

examples/get_started_fasterrcnn.py

@@ -159,11 +159,23 @@ def plot_image_with_boxes(img, boxes, pred_cls):
     for i in range(len(boxes)):
         # Draw Rectangle with the coordinates
 
-        cv2.rectangle(img, (int(boxes[i][0][0]), int(boxes[i][0][1])), (int(boxes[i][1][0]), int(boxes[i][1][1])),
-                      color=(0, 255, 0), thickness=rect_th)
+        cv2.rectangle(
+            img,
+            (int(boxes[i][0][0]), int(boxes[i][0][1])),
+            (int(boxes[i][1][0]), int(boxes[i][1][1])),
+            color=(0, 255, 0),
+            thickness=rect_th,
+        )
         # Write the prediction class
-        cv2.putText(img, pred_cls[i], (int(boxes[i][0][0]), int(boxes[i][0][1])), cv2.FONT_HERSHEY_SIMPLEX, text_size,
-                    (0, 255, 0), thickness=text_th)
+        cv2.putText(
+            img,
+            pred_cls[i],
+            (int(boxes[i][0][0]), int(boxes[i][0][1])),
+            cv2.FONT_HERSHEY_SIMPLEX,
+            text_size,
+            (0, 255, 0),
+            thickness=text_th,
+        )
     plt.axis("off")
     plt.imshow(img.astype(np.uint8), interpolation="nearest")
     plt.show()
@@ -281,7 +293,7 @@ def append_loss_history(loss_history, output):
             file.write(json.dumps(loss_history))
 
         np.save(os.path.join(config["path"], "patch"), attack._patch)
-        
+
     predictions_adv = frcnn.predict(x=x_patch)
 
     for i in range(image.shape[0]):
@@ -291,4 +303,4 @@ def append_loss_history(loss_history, output):
         predictions_adv_class, predictions_adv_boxes, predictions_adv_class = extract_predictions(predictions_adv[i])
 
         # Plot predictions
-        plot_image_with_boxes(img=x_patch[i].copy(), boxes=predictions_adv_boxes, pred_cls=predictions_adv_class)
+        plot_image_with_boxes(img=x_patch[i].copy(), boxes=predictions_adv_boxes, pred_cls=predictions_adv_class)