merged with main

Signed-off-by: Peter Staar <[email protected]>

tests/test_layout_predictor.py

@@ -4,6 +4,7 @@
 #
 import os
 import json
+from pathlib import Path
 
 import torch
 import numpy as np
@@ -74,19 +75,17 @@ def test_layoutpredictor(init: dict):
 
         true_layout_fn = img_fn+".json"
         with Image.open(img_fn) as img:
-            pred_layout=[]
-
+
+            w, h = img.size
+
             # Load images as PIL objects
             for i, pred in enumerate(lpredictor.predict(img)):
                 print("PIL pred: {}".format(pred))
-                pred_layout.append({
-                    "label": pred["label"],
-                    "t": pred["t"].item(),
-                    "b": pred["b"].item(),
-                    "l": pred["l"].item(),
-                    "r": pred["r"].item(),
-                })
-            print(pred_layout)
+                assert pred["l"] >= 0 and pred["l"] <= w
+                assert pred["t"] >= 0 and pred["t"] <= h
+                assert pred["r"] >= 0 and pred["r"] <= w
+                assert pred["b"] >= 0 and pred["b"] <= h
+
             assert i + 1 == init["pred_bboxes"]
 
             if os.path.exists(true_layout_fn):