We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2173de8 commit d022b49Copy full SHA for d022b49
src/image/image.py
@@ -1,4 +1,5 @@
1
import io
2
+from typing import Optional
3
4
import numpy as np
5
import torch
@@ -18,7 +19,12 @@ def get_tensor_from_np_array(np_array: np.ndarray) -> torch.Tensor:
18
19
20
21
class ImageObject:
- def __init__(self, viam_image: ViamImage, pil_image: Image = None, device=None):
22
+ def __init__(
23
+ self,
24
+ viam_image: ViamImage,
25
+ pil_image: Optional[Image.Image] = None,
26
+ device=None
27
+ ):
28
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
29
if pil_image is not None:
30
self.pil_image = pil_image
0 commit comments