mirror of
https://github.com/Stirling-Tools/Stirling-PDF.git
synced 2026-09-03 05:10:16 +03:00
Fix prettier and ruff formatting for form detection files
This commit is contained in:
+28
-168
@@ -5,213 +5,73 @@
|
||||
"queries": [
|
||||
{
|
||||
"q": 0,
|
||||
"dets": [
|
||||
0.594326,
|
||||
0.253922,
|
||||
0.575291,
|
||||
0.03704
|
||||
],
|
||||
"labels": [
|
||||
1.494305,
|
||||
-4.254563,
|
||||
-2.83488,
|
||||
-8.221176
|
||||
]
|
||||
"dets": [0.594326, 0.253922, 0.575291, 0.03704],
|
||||
"labels": [1.494305, -4.254563, -2.83488, -8.221176]
|
||||
},
|
||||
{
|
||||
"q": 1,
|
||||
"dets": [
|
||||
0.593039,
|
||||
0.305142,
|
||||
0.574628,
|
||||
0.036171
|
||||
],
|
||||
"labels": [
|
||||
1.501225,
|
||||
-4.211613,
|
||||
-2.82295,
|
||||
-8.282976
|
||||
]
|
||||
"dets": [0.593039, 0.305142, 0.574628, 0.036171],
|
||||
"labels": [1.501225, -4.211613, -2.82295, -8.282976]
|
||||
},
|
||||
{
|
||||
"q": 2,
|
||||
"dets": [
|
||||
0.594104,
|
||||
0.355328,
|
||||
0.573003,
|
||||
0.036946
|
||||
],
|
||||
"labels": [
|
||||
1.550975,
|
||||
-4.177611,
|
||||
-3.121929,
|
||||
-8.416718
|
||||
]
|
||||
"dets": [0.594104, 0.355328, 0.573003, 0.036946],
|
||||
"labels": [1.550975, -4.177611, -3.121929, -8.416718]
|
||||
},
|
||||
{
|
||||
"q": 3,
|
||||
"dets": [
|
||||
0.593218,
|
||||
0.204408,
|
||||
0.573657,
|
||||
0.036336
|
||||
],
|
||||
"labels": [
|
||||
1.483687,
|
||||
-4.080855,
|
||||
-2.717842,
|
||||
-8.144939
|
||||
]
|
||||
"dets": [0.593218, 0.204408, 0.573657, 0.036336],
|
||||
"labels": [1.483687, -4.080855, -2.717842, -8.144939]
|
||||
},
|
||||
{
|
||||
"q": 4,
|
||||
"dets": [
|
||||
0.593249,
|
||||
0.102776,
|
||||
0.573464,
|
||||
0.03605
|
||||
],
|
||||
"labels": [
|
||||
1.52258,
|
||||
-4.080616,
|
||||
-2.611064,
|
||||
-8.24098
|
||||
]
|
||||
"dets": [0.593249, 0.102776, 0.573464, 0.03605],
|
||||
"labels": [1.52258, -4.080616, -2.611064, -8.24098]
|
||||
},
|
||||
{
|
||||
"q": 5,
|
||||
"dets": [
|
||||
0.593204,
|
||||
0.153512,
|
||||
0.572868,
|
||||
0.036364
|
||||
],
|
||||
"labels": [
|
||||
1.489175,
|
||||
-4.014163,
|
||||
-2.746595,
|
||||
-8.259592
|
||||
]
|
||||
"dets": [0.593204, 0.153512, 0.572868, 0.036364],
|
||||
"labels": [1.489175, -4.014163, -2.746595, -8.259592]
|
||||
},
|
||||
{
|
||||
"q": 6,
|
||||
"dets": [
|
||||
0.593207,
|
||||
0.406301,
|
||||
0.575204,
|
||||
0.036417
|
||||
],
|
||||
"labels": [
|
||||
1.523296,
|
||||
-4.076083,
|
||||
-2.997732,
|
||||
-8.246349
|
||||
]
|
||||
"dets": [0.593207, 0.406301, 0.575204, 0.036417],
|
||||
"labels": [1.523296, -4.076083, -2.997732, -8.246349]
|
||||
},
|
||||
{
|
||||
"q": 7,
|
||||
"dets": [
|
||||
0.336551,
|
||||
0.487273,
|
||||
0.021658,
|
||||
0.016205
|
||||
],
|
||||
"labels": [
|
||||
-2.577461,
|
||||
1.216892,
|
||||
-4.798167,
|
||||
-7.703153
|
||||
]
|
||||
"dets": [0.336551, 0.487273, 0.021658, 0.016205],
|
||||
"labels": [-2.577461, 1.216892, -4.798167, -7.703153]
|
||||
},
|
||||
{
|
||||
"q": 8,
|
||||
"dets": [
|
||||
0.824501,
|
||||
0.570058,
|
||||
0.114489,
|
||||
0.03537
|
||||
],
|
||||
"labels": [
|
||||
1.210955,
|
||||
-3.528949,
|
||||
-3.637566,
|
||||
-8.277011
|
||||
]
|
||||
"dets": [0.824501, 0.570058, 0.114489, 0.03537],
|
||||
"labels": [1.210955, -3.528949, -3.637566, -8.277011]
|
||||
},
|
||||
{
|
||||
"q": 9,
|
||||
"dets": [
|
||||
0.499909,
|
||||
0.487268,
|
||||
0.021303,
|
||||
0.016313
|
||||
],
|
||||
"labels": [
|
||||
-2.650476,
|
||||
1.275709,
|
||||
-4.80449,
|
||||
-7.729639
|
||||
]
|
||||
"dets": [0.499909, 0.487268, 0.021303, 0.016313],
|
||||
"labels": [-2.650476, 1.275709, -4.80449, -7.729639]
|
||||
},
|
||||
{
|
||||
"q": 10,
|
||||
"dets": [
|
||||
0.480432,
|
||||
0.569626,
|
||||
0.345508,
|
||||
0.03706
|
||||
],
|
||||
"labels": [
|
||||
-0.667743,
|
||||
-4.025378,
|
||||
-0.391272,
|
||||
-7.142419
|
||||
]
|
||||
"dets": [0.480432, 0.569626, 0.345508, 0.03706],
|
||||
"labels": [-0.667743, -4.025378, -0.391272, -7.142419]
|
||||
},
|
||||
{
|
||||
"q": 99,
|
||||
"dets": [
|
||||
0.787065,
|
||||
0.572109,
|
||||
0.041269,
|
||||
0.032919
|
||||
],
|
||||
"labels": [
|
||||
-4.071043,
|
||||
-4.381146,
|
||||
-5.294834,
|
||||
-8.117142
|
||||
]
|
||||
"dets": [0.787065, 0.572109, 0.041269, 0.032919],
|
||||
"labels": [-4.071043, -4.381146, -5.294834, -8.117142]
|
||||
},
|
||||
{
|
||||
"q": 150,
|
||||
"dets": [
|
||||
0.617099,
|
||||
0.154866,
|
||||
0.457747,
|
||||
0.034707
|
||||
],
|
||||
"labels": [
|
||||
-3.982041,
|
||||
-5.252909,
|
||||
-4.907092,
|
||||
-8.302184
|
||||
]
|
||||
"dets": [0.617099, 0.154866, 0.457747, 0.034707],
|
||||
"labels": [-3.982041, -5.252909, -4.907092, -8.302184]
|
||||
},
|
||||
{
|
||||
"q": 299,
|
||||
"dets": [
|
||||
0.595358,
|
||||
0.708946,
|
||||
0.331338,
|
||||
0.039393
|
||||
],
|
||||
"labels": [
|
||||
-4.61758,
|
||||
-5.568352,
|
||||
-4.882038,
|
||||
-7.22629
|
||||
]
|
||||
"dets": [0.595358, 0.708946, 0.331338, 0.039393],
|
||||
"labels": [-4.61758, -5.568352, -4.882038, -7.22629]
|
||||
}
|
||||
],
|
||||
"expected": [
|
||||
|
||||
@@ -56,9 +56,7 @@ def fetch_checkpoint(dest: Path) -> Path:
|
||||
|
||||
actual = sha256_of(dest)
|
||||
if actual != CHECKPOINT_SHA256:
|
||||
raise SystemExit(
|
||||
f"checkpoint sha256 mismatch\n expected {CHECKPOINT_SHA256}\n actual {actual}"
|
||||
)
|
||||
raise SystemExit(f"checkpoint sha256 mismatch\n expected {CHECKPOINT_SHA256}\n actual {actual}")
|
||||
print(f"checkpoint verified ({dest.stat().st_size / 1e6:.1f} MB)")
|
||||
return dest
|
||||
|
||||
@@ -111,9 +109,7 @@ def verify(model: Path) -> None:
|
||||
if names != ["dets", "labels"]:
|
||||
raise SystemExit(f"unexpected output names {names}; the decoder binds by name")
|
||||
|
||||
outputs = session.run(
|
||||
None, {session.get_inputs()[0].name: np.zeros((1, 3, INPUT_SIZE, INPUT_SIZE), np.float32)}
|
||||
)
|
||||
outputs = session.run(None, {session.get_inputs()[0].name: np.zeros((1, 3, INPUT_SIZE, INPUT_SIZE), np.float32)})
|
||||
for name, value in zip(names, outputs):
|
||||
if np.isnan(value).any():
|
||||
raise SystemExit(f"output {name} contains NaN")
|
||||
|
||||
Reference in New Issue
Block a user