mirror of
https://github.com/Refound-445/nonebot-plugin-nailongremove.git
synced 2025-11-04 21:22:43 +08:00
This commit is contained in:
parent
480667ecc5
commit
dea6baa13e
@ -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,
|
||||
|
||||
Loading…
Reference in New Issue
Block a user