We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a1b7406 commit 1fa22b7Copy full SHA for 1fa22b7
src/mock_vws/target_raters.py
@@ -8,7 +8,7 @@
8
9
import piq # type: ignore[import-untyped]
10
from PIL import Image
11
-from torchvision.transforms import functional # type: ignore[import-untyped]
+from torchvision import transforms
12
13
14
@functools.cache
@@ -25,7 +25,8 @@ def _get_brisque_target_tracking_rating(image_content: bytes) -> int:
25
"""
26
image_file = io.BytesIO(initial_bytes=image_content)
27
image = Image.open(fp=image_file)
28
- image_tensor = functional.to_tensor(pic=image) * 255 # pyright: ignore[reportUnknownMemberType]
+ transform = transforms.ToTensor()
29
+ image_tensor = transform(pic=image)
30
image_tensor = image_tensor.unsqueeze(0)
31
try:
32
brisque_score = piq.brisque(x=image_tensor, data_range=255)
0 commit comments