up
Some checks failed
Pre Commit Check / pre-commit (push) Has been cancelled

This commit is contained in:
Refound-445 2024-12-09 15:58:46 +08:00
parent 480667ecc5
commit dea6baa13e

View File

@ -49,7 +49,6 @@ def check_single(image: np.ndarray, is_gif: bool = False) -> CheckSingleResult[N
image = image.unsqueeze(0) # type: ignore
with torch.no_grad():
output = model(image.to(device)) # type: ignore
print(output)
_, pred = torch.max(output, 1)
return CheckSingleResult(
ok=pred.item() == 1,